安装GPU的可以参考 https://www.cnblogs.com/codeit/p/13253449.html
最近了解并尝试在Win10安装YOLOv3,参考了十几篇文章,发现每个人都有自己的安装方式,最初尝试用cmake编译,虽然安装完成,坑非常多,经2天努力终于安装成功,分享并记录自己的安装过程,供大家参考。
系统:Win10
显卡:GeForce RTX2070 with Max-Q Design
其它:CUDA10.2,cuDNN7.6.5,VS2019,OpenCV3.4.0(opencv版本最好别高于3.4.0)
VS2019下载和安装
https://visualstudio.microsoft.com/zh-hans/downloads
在这里插入图片描述
选择“.Net桌面开发”、“使用C++的桌面开发”、“通用Windwos平台开发”三项,点击右下角安装:
YOLO下载和编译加粗样式****
点击https://github.com/AlexeyAB/darknet 下载YOLO。添加链接描述
这里千万要下对了。。。。
复制编译依赖文件
将c:/opencv/build/x64/vc15/bin目录下的opencv_ffmpeg340_64.dll、opencv_world340.dll复制到D:\darknet\build\darknet\x64目录下:
进入D:\darknet\build\darknet目录,用VS2019打开darknet.sln(这个是用显卡才编辑这个),如果不用显卡是编辑darknet_no_gpu.sln,这个非常重要。
首次打开可能会出现以下错误:
没关系,确定后,关闭VS2019,重新用VS2019打开darknet.sln即可,打开后此处选择Release X64。
![在这里插入图片描述](https://img-blog.csdnimg.cn/20201016170046551.png#pic_center
右键darknet工程,选择属性。
将“常规”下的Windows SDK版本选为10.0,将平台工具集选为Visual Studio2019(v142),点击应用。
然后点击VC++目录,编辑包含目录,将以下路径添加到包含目录中:
C:\opencv\build\include\opencv2
C:\opencv\build\include\opencv
C:\opencv\build\include
opencv路径请修改成你自己的路径。
然后点击库目录,把以下内容添加到库目录中:
C:\opencv\build\x64\vc14\lib
然后点击链接器->输入,在附加依赖项中添加:
C:\opencv\build\x64\vc14\lib\opencv_world340.lib
然后点击确定保存,并退出VS2019。
用VS2019打开darknet.sln,再次确保此处为Release X64,然后右键点击项目选择生成。
编译过程可能会出现很多警告,但只要最终提示生成:成功1个,失败0个,则表示编译成功。
如果上面的都编辑了还是失败可以编辑C++
下载YOLOv3预训练权重文件
进入https://pjreddie.com/media/files/yolov3.weights
下载yolov3.weights文件,再将yolov3.weights放在D:\darknet\build\darknet\x64目录下。
如果下载慢,选择迅雷下载。
然后是 运行YOLO
控制台进入D:\darknet\build\darknet\x64目录,输入命令:
目录是自己的目录
darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
效果还是挺好的,我只用100张图片迭代了600次就有这么高的识别率了。
loss 下降到了0.6
注意事项
darkent.sln目录下有两个这种DLL文件,如果要用python的话还是推荐也编译一下,按照上面的一样。添加那些路径。
不然会出现这种错误