天天看点

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未解决

继续阅读