yolov5,训练后 效果特差

问题描述:

第一次训练后,根本没有结果。经过更改,我把所有图片统一更改为800×600,train的image-size 设定为800×800,但是训练效果仍然很差(如下图,所有的训练结果),求助各位大佬

 

 

几个问题:

1.所有图片统一更改为800×600:你是直接resize的吗还是按照比例缩放然后填充的呢?这个影响巨大!还有,作者有说过,训练的图片最好的32的整数倍,所以我建议你还是不要更改,然后使用原来的设置640*640。如果你觉得640太小的,可以增加,但是最好长宽都是32的整数倍,不需要自己修改图片的大小,yolo本身自带自适应缩放,只要设置下输入大小就行,但是请不要使用opencv的risize函数来更改图片大小,这样子会使得物体形变。

2.你的数据集是多少?每个物体的个数是多少?这个影响也很大,你的验证集和训练集的loss都在下降,但是你的测试效果和你的map曲线很低,怀疑你的数据集不够大,也就是泛化能力不行,只能针对数据集的数据进行检测,而数据集之外的数据就不行了。

3.从你的训练图片来看,你的100个epoch有点少了,看不到收敛的迹象,建议增大epoch。