yolov5,训练后 效果特差

Question

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

 

 

Answer 1

几个问题：

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

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

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