跑通YOLO-Fastest

跑通YOLO-Fastest

一、代码来源

https://github.com/dog-qiuqiu/Yolo-Fastest

二、环境配置

1.win10
2.cuda10.2+cudnn
3.opencv451
4.vs2015
5.cmake

三、编译代码

1.修改Makefile配置

GPU=1
CUDNN=1
CUDNN_HALF=0
OPENCV=1
AVX=1
OPENMP=0
LIBSO=0
ZED_CAMERA=0
ZED_CAMERA_v2_8=0

主要根据自己的环境设置GPU=1；CUDNN=1；CUDNN_HALF=0；OPENCV=1四项，1为启用。
2.Cmake
1)打开Cmake，配置

资源和目标路径都指定到yolo-fastest文件夹。
2）点击configure,选用vs2015和x64平台。
3）点击finish既可。
4）点击generate
注：提示opencv错误，但是实际上电脑配置了opencv时，可以修改Environment中的opencv路径

3.vs2015
1)打开下面ALL_BUILD.vcxproj

生成既可

成功后会生成Release文件夹

打开Release文件夹，把里面编译完成的darknet.dll和darknet.exe复制到Yolo-Fastest/build/darknet/x64目录下；
把ModelZoo\yolo-fastest-1.1_coco目录下的4个文件复制到Yolo-Fastest/build/darknet/x64/cfg目录下；

四、测试

build/darknet/x64文件夹下打开命令行窗口
图片测试

 darknet.exe detector test ./cfg/coco.data ./cfg/yolo-fastest-1.1.cfg ./cfg/yolo-fastest-1.1.weights ./data/person.jpg 

结果

注意：这里yolo-fastest-1.1.cfg 和yolo-fastest-1.1.weights要改成自己选用的。
视频测试

 darknet.exe detector demo ./cfg/coco.data ./cfg/yolo-fastest-1.1.cfg ./cfg/yolo-fastest-1.1.weights ./data/test_car.mp4

需要下载对应视频