Light version of convolutional neural network Yolo v2 for objects detection with a minimum of dependencies
This repository supports:
- both Windows and Linux
- both OpenCV 3.x and OpenCV 2.4.13
- both cuDNN 5 and cuDNN 6
- CUDA >= 7.5
How to compile:
-
To compile for CPU just do
make
on Linux or buildyolo_cpu.sln
on Windows -
To compile for GPU set flag
GPU=1
in theMakefile
on Linux or buildyolo_gpu.sln
on WindowsRequired both CUDA >= 7.5 and cuDNN >= 5.1
How to start:
- Download
yolo-voc.weights
to thebin
directory and run./yolo.sh
on Linux (oryolo_cpu.cmd
/yolo_gpu.cmd
on Windows) - Download
tiny-yolo-voc.weights
to thebin
directory and run./tiny-yolo.sh
Other models by the link: https://pjreddie.com/darknet/yolo/