最近心血來潮想學一下YOLOv3,於是就去網上看了YOLOv3在win10下的配置教程。在配置過程中塌坑無數,花了很多時間和精力,所以我想就此寫一篇博客來介紹在在win10+vs2015的環境下如何配置YOLOv3。
1.安裝VS(YOLOv3支持vs2012、2013、2015,其他版本需要下載工具集)
儘量安裝vs2015
2.安裝CUDA
2.1確定自己電腦對應的CUDA版本號
2.1.1打開控制面板
2.1.2進入NVIDIA控制面板
2.1.3選擇系統信息
2.1.4點擊組件,即可看到自己電腦適應的CUDA的版本號
2.2進入官網https://developer.nvidia.com/cuda-downloads ,按照自己的系統選擇合適的版本。
3.安裝CUDNN
CUDNN的版本需要與CUDA版本對應,下載的時候根據選定的CUDA版本進行選擇即可。
下載鏈接: https://developer.nvidia.com/cudnn
下載之後把對應文件夾的文件直接放入CUDA的對應文件夾就可以了
即
將D:\安裝包\cudnn-9.1-windows10-x64-v7.1\cuda\bin\cudnn64_7.dll 複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
將 D:\安裝包\cudnn-9.1-windows10-x64-v7.1\cudacuda\ include\cudnn.h 複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\include
將D:\安裝包\cudnn-9.1-windows10-x64-v7.1\cudacuda\lib\x64\cudnn.lib複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\lib\x64
複製完之後要去檢查一下cuda和cudnn的環境變量 ,務必要保證這些環境變量都有要不然最後在測試時會報錯
4. 安裝 OpenCV (OpenCV 2.x.x 或者 OpenCV <= 3.4.0)
(切記opencv的版本不能超過3.40,我當時就是因爲粗心沒注意到這點才折騰了很久)
下載地址:https://sourceforge.net/projects/opencvlibrary/files/opencvwin/3.4.0/opencv-3.4.0-vc14_vc15.exe/download
如果訪問不了,可以使用我的鏈接
鏈接: https://pan.baidu.com/s/1yACjz1AiDWtkUzxZHqCH0w 提取碼: scxv
接下來將opencv配置到環境變量
5. 下載darknet
下載地址:https://github.com/AlexeyAB/darknet
(如果會使用git,則直接git clone https://github.com/AlexeyAB/darknet)
5.1修改darknet.vcxproj文件
進入darknet\build\darknet目錄,用記事本打開darknet.vcxproj,將所有CUDA 10.0修改爲自己對應的CUDA版本,本篇使用CUDA 9.1,故改爲9.1。(共有兩處需要修改)
5.2打開darknet.sln
5.2.1將項目修改爲Release x64
5.2.2配置包含目錄 + 庫目錄 + 鏈接器
包含目錄:在darknet項目上點擊鼠標右鍵->屬性,彈出如下界面:然後VC++目錄-->包含目錄-->編輯
添加的目錄:...\opencv\build\include(...代表opencv的安裝路徑,下文同理)
...\opencv\build\include\opencv
...\opencv\build\include\opencv2
庫目錄:方法與包含目錄類似,添加的目錄爲:...\opencv\build\x64\vc14\lib
鏈接器:添加目錄...\opencv\build\x64\vc14\lib下庫的名字:opencv_world340.lib
5.2.3右鍵darknet,選擇屬性,進入CUDA C/C++中的Code Generation,刪掉compute_75,sm_75(此步極爲關鍵,要不然必出錯)
5.3右鍵darknet,選擇生成
編譯成功,會在darknet\build\darknet\x64下得到darknet.exe
當你看到darknet.exe時,那麼就恭喜你配置成功過了!
6.測試
6.1下載yolov3.weights
官網:https://pjreddie.com/media/files/yolov3.weights
下載速度有點慢
也可以用我的鏈接 鏈接: https://pan.baidu.com/s/1FSkXMGqCeqDFpeDxJzb3sA 提取碼: 1ajt
下載完成後把它放到darknet\build\darknet\x64下
6.2測試
在darknet\build\darknet\x6下按住shift鍵,點擊鼠標右鍵選擇“在此處打開Powershell 窗口(s)”
在命令行輸入 .\darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
如果看到以下結果,說明你成功的配好了YOLOv3
原文出處:https://www.cnblogs.com/songhouhou/p/10805976.html