當前位置: 華文頭條 > 推薦

物體分割檢測YOLO4演算法環境配置

2024-01-24推薦

安裝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