三步搞定:
1、下载darknet:
https://github.com/pjreddie/darknet
2、更改makefile然后make
cpu设置:gpu=0,opencv=1,cudnn=0
gpu设置:gpu=1,opencv=1,cudnn=1
cd darknet
make
3、下载训练好的模型
http://pjreddie.com/media/files/yolo.weights
然后run一副图像就好了,现在用data中的一副图像:
./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg
看一下效果:
当然你可以自己定义图片的路径:
先输入:
./darknet detect cfg/yolo.cfg yolo.weights
然后再输入图片的路径:
data/horses.jpg
可以修改训练图像的阈值:
./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg -thresh 0
0.05的阈值情况下: