之前文章中写过关于在windows环境下搭建yolov5的文章,也写过如何利用yolov5训练自己的数据集,总的来说yolov5还是很便捷的,但是在实际的工程问题上,yolov5存在落地打包上的困难,因此本文记录下yolov4在windows环境下搭建。
系列文章目录
安装依赖
括号里是我当前版本
vs(vs2019)
cuda(cuda10.2)
cudnn(cudnn7.6.5)
opencv(opencv4.5.0)
cmake(cmake3.19)
编译
打开cmake,按自己所需设置yolov4源码的位置,和mybuild的位置:
然后点击Configure
,之后按照下图进行设置:
其中ENABLE_CUDNN_HALF
按需设置
然后在此点击Configure
进行配置,待其配置完成之后,点击Generat
e进行生成。
待其Generate done
之后,进入设置mybuild
目录,选中*.sln
,用VS进行打开,会有如下提示,记得点击确定。
之后选择Release x64
配置运行编译
使用步骤
进入Release
目录下,将项目跟目录下的data
文件夹、cfg
文件夹和3rdparty\pthreads\bin\pthreadVC2.dll
和下载下来的权重文件放到Release
目录下
视频检测命令:
darknet.exe detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights 001.avi -i 0 -thresh 0.25
图片检测命令:
darknet.exe detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights data/dog.jpg -thresh 0.25
调起摄像头:
darknet.exe detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights -c 0
总结
yolov4图片测试结果:
参考文档:
https://blog.csdn.net/u014587123/article/details/112427025
https://blog.csdn.net/weixin_40841247/article/details/105742022
如果阅读本文对你有用,欢迎关注点赞评论收藏呀!!!
2021年3月8日10:47:41