windows10+vs2017+opencv3.4.1配置YOLOV4

首先需要準備的軟件包:我這裏CUDA是10.0版本,CUDNN是對應10.0的版本,一定要對應,可以在控制面板看一下自己的CUDA是什麼版本。

1.CUDA:https://developer.nvidia.com/cuda-toolkit-archive

2.CUDNN:https://developer.nvidia.com/rdp/cudnn-archive

3.opencv>2.4

4.cmake>=3.8:https://cmake.org/download/

1.下載完CUDA和CUDNN之後,CUDA默認安裝在C盤:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

此時,解壓CUDNN,按照下圖路徑在解壓後的安裝包內找到.dll,.h和.lib,複製到對應的CUDA目錄下。然後要將CUDA添加到系統環境變量裏面

2.下載YOLOV4源碼:https://github.com/AlexeyAB/darknet然後打開CMAKE-GUI

上圖第一行是你下載源碼後的保存的地方,第二行是你要編譯生成放置的地方,然後點擊Configure,選擇平臺爲visual studio15 2017和x64,接着再進行下圖的配置:注意第二行的OPENCV_DIR是目前電腦上安裝opencv的路徑

接着點擊gengerate,出現done就表明生成完成

3.生成後用VS2017打開Darknet.sln,選擇平臺爲release,x64,然後右鍵點擊生成

4.生成後進入release目錄下,將源碼中的data文件夾cfg文件夾3rdparty\pthreads\bin\pthreadVC2.dll和下載下來的yolov4.weight權重文件放到Release目錄

yolov4.weight下載地址:https://drive.google.com/uc?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT&export=download

然後在release目錄下,按住shift鍵,再點擊鼠標右鍵,選擇在powershell中運行,輸入以下命令:

.\darknet.exe detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights -c 0

回車運行,就會打開系統攝像頭和檢測結果,如下圖所示.

此時代表windows10,YOLOV4配置成功

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