YOLO V4學習記錄
一、安裝問題
顯示卡:GTX 2080SUPER;CUDA:10.2;windows10專業版
按照網絡上提供的教程安裝、配置YOLO V4,安裝CUDA 10.2軟體,安裝對用cuDNN版本,完成所有編譯,最後測試無法使用GPU加速
解決辦法一:按照原作者使用的CUDA 10.0版本重新編譯,對用cuDNN版本,仍然無法使用GPU加速。
解決辦法二:安裝CUDA 10.1版本編譯,對應cuDNN版本,仍然無法使用GPU加速。
解決辦法三:安裝CUDA 10.2版本編譯,對應cuDNN版本,可以使用GPU加速,完成安裝。
二、資料集訓練問題
訓練參考https://blog.csdn.net/weixin_39954922/article/details/105816571
1、訓練時顯示CUDA Error: out of memory darknet。
解決辦法:修改所使用的模型cfg檔案中的subdivision的參數。由subdivisions=8改成subdivisions=64。
2、訓練資料時一切正常, 用訓練好的模型測試圖像出現如下情況:

圖一:損失曲線圖 圖二:預測圖
解決辦法:未解決!!!
3、訓練資料時在輸入指令中加入想在損失曲線圖中顯示mAP的變化曲線。
解決辦法:-map: .\darknet.exe detector train data/Captainjsong.data cfg/yolov4-Captainjsong.cfg yolov4.conv.137 -map未解決