首先需要準備的軟件包:我這裏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配置成功