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、训练数据时一切正常， 用训练好的模型测试图像出现如下情况：

                  

                        图一：损失曲线图                                                                图二：预测图 

解决办法：未解决！！！ 

3、训练数据时在输入指令中加入想在损失曲线图中显示mAP的变化曲线。

解决办法：-map： .\darknet.exe detector train data/Captainjsong.data cfg/yolov4-Captainjsong.cfg yolov4.conv.137 -map未解决