YOLOv3配置(win10+opencv3.40+cuda9.1+cudnn7.1+vs2015)

 

最近心血來潮想學一下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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章