安装

下载 yolo

git clone https://github.com/pjreddie/darknet

修改训练结果输出频率

在文件 examples/detector.c中的第138行

if(i%10000==0 || (i < 1000 && i%100 == 0)){

把10000改成1000

if(i%1000==0 || (i < 1000 && i%100 == 0)){

设置开启gpu,opencv等

需要事先安装 cuda opencv
打开Makefile,修改前几行

GPU=1
CUDNN=1
OPENCV=1
OPENMP=1

编译

cd darknet
make
sudo ln -s $(pwd)/darknet /usr/local/bin/darknet

演示

下载coco权重文件

wget https://pjreddie.com/media/files/yolov3.weights

识别图片

darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

改变阈值

在命令后面加上 -thresh 0.5

识别视频 或者 实时webcam

darknet detector demo cfg/coco.data cfg/yolov3.cfg yolov3.weights <video file>

如果不加 <video file>则识别实时webcam