天天看點

YOLO V4學習記錄

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、訓練資料時一切正常, 用訓練好的模型測試圖像出現如下情況:

YOLO V4學習記錄
YOLO V4學習記錄

                        圖一:損失曲線圖                                                                圖二:預測圖 

解決辦法:未解決!!! 

3、訓練資料時在輸入指令中加入想在損失曲線圖中顯示mAP的變化曲線。

解決辦法:-map: .\darknet.exe detector train data/Captainjsong.data cfg/yolov4-Captainjsong.cfg yolov4.conv.137 -map未解決

繼續閱讀