Download the Models

Run the getModels.sh file from command line to download the needed model files

sudo chmod a+x getModels.sh
./getModels.sh

How to run the code

Command line usage for object detection using YOLOv3

• Python

  • Using CPU

    • A single image:

    python3 object_detection_yolo.py --image=bird.jpg --device 'cpu'
    
    • A video file:

    python3 object_detection_yolo.py --video=run.mp4 --device 'cpu'
    

  • Using GPU

    • A single image:

    python3 object_detection_yolo.py --image=bird.jpg --device 'gpu'
    
    • A video file:

    python3 object_detection_yolo.py --video=run.mp4 --device 'gpu'
    

• C++:

  • Using CPU

    • A single image:

    ./build/object_detection_yolo --image=bird.jpg --device=cpu
    
    • A video file:

     ./build/object_detection_yolo --video=run.mp4 --device=cpu
    

  • Using GPU

    • A single image:

    ./build/object_detection_yolo --image=bird.jpg --device=gpu
    
    • A video file:

     ./build/object_detection_yolo --video=run.mp4 --device=gpu
    

Compilation examples

• Using g++

g++ -ggdb pkg-config --cflags --libs /usr/local/Cellar/opencv3/3.4.2/lib/pkgconfig/opencv.pc object_detection_yolo.cpp -o object_detection_yolo.out

• Using CMake

  • On Unix systems

  mkdir build && cd build
  cmake ..
  cmake --build . --config Release
  cd ..
  
  • On Windows systems

  mkdir build
  cd build
  cmake -G "Visual Studio 16 2019" ..
  cmake --build . --config Release
  cd ..
  

Note: To run on Windows system, change syntax accordingly:

.\build\Release\object_detection_yolo --video=run.mp4 --device=gpu

Results of YOLOv3

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.