Skip to content

Real time object detection with deployment of YOLOv5 through LibTorch C++ API

License

Notifications You must be signed in to change notification settings

Nebula4869/YOLOv5-LibTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YOLOv5-LibTorch

Real time object detection with deployment of YOLOv5 through LibTorch C++ API

Environment

  • Ubuntu 18.04
  • OpenCV 3.2.0
  • LibTorch 1.6.0
  • CMake 3.10.2

Getting Started

  1. Install OpenCV.

    sudo apt-get install libopencv-dev
  2. Install LibTorch.

    wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
    unzip libtorch-shared-with-deps-latest.zip
  3. Edit "CMakeLists.txt" to configure OpenCV and LibTorch correctly.

  4. Compile and run.

    cd build
    cmake ..
    make
    ./../bin/YOLOv5LibTorch

Note: COCO-pretrained YOLOv5s model has been provided. For more pretrained models, see yolov5.

About

Real time object detection with deployment of YOLOv5 through LibTorch C++ API

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published