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

 

最近心血來潮想學一下YOLOv3,因而就去網上看了YOLOv3在win10下的配置教程。在配置過程當中塌坑無數,花了不少時間和精力,因此我想就此寫一篇博客來介紹在在win10+vs2015的環境下如何配置YOLOv3。git

1.安裝VS(YOLOv3支持vs20十二、201三、2015,其餘版本須要下載工具集)

儘可能安裝vs2015github

2.安裝CUDA

2.1肯定本身電腦對應的CUDA版本號shell

2.1.1打開控制面板windows

2.1.2進入NVIDIA控制面板工具

 

2.1.3選擇系統信息

測試

2.1.4點擊組件,便可看到本身電腦適應的CUDA的版本號ui

 

 2.2進入官網https://developer.nvidia.com/cuda-downloads ,按照本身的系統選擇合適的版本。spa

3.安裝CUDNN

CUDNN的版本須要與CUDA版本對應,下載的時候根據選定的CUDA版本進行選擇便可。.net

下載連接:  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

相關文章
相關標籤/搜索