安裝Opencv
opencv網址: https://opencv.org/releases/
環境變量配置
系統變量:OpenCV_DIR D:\BlackCatSoftWare\opencv\opencv
再進入Path,添加:
D:\BlackCatSoftWare\opencv\opencv\build\x64\vc16\binD:\BlackCatSoftWare\opencv\opencv\build\x64\vc16\lib
安裝 CUDA
cuda下載: https://developer.nvidia.com/cuda-toolkit-archive
檢視適合的cuda版本:
我的系統支持10.2版本。
自訂,全部勾選:
檢測安裝成功:nvcc -V
安裝cuDNN
cuDNN網址: https://developer.nvidia.com/rdp/cudnn-archive
需要註冊賬號登入後才可以下載。
將解壓後的三個資料夾復制到Cuda對應目錄下。
安裝CMAKE
cmake網址: https://cmake.org/download/
安裝成功:
編譯darknet
git clone https://github.com/AlexeyAB/darknet.git
Path中添加環境變量:C:\Users\Cat\Desktop\darknet\build\darknet\x64
下載權重檔yolov4.weights放在darknet目錄下
修改檔,將cuda改為目前版本號:
<ImportGroup Label="ExtensionSettings"> <Import Project="$(VCTargetsPath)\BuildCustomizations\CUDA 10.2.props" /></ImportGroup><ImportGroup Label="ExtensionTargets"> <Import Project="$(VCTargetsPath)\BuildCustomizations\CUDA 10.2.targets" /></ImportGroup>
VS2022開啟專案:
1.將上方解決方案配置和解決方案平台更改為Release和x64
2.引入opencv
windows上進行訓練:darknet.exe detector train data/obj.data data/yolo-obj.cfg data/yolov4.conv.137 -map