DataXujing/YOLO-v5

使用单张图片训练和测试，模型为什么没办法拟合数据集，检测不出目标，mAP很小，求大神解答

jingenyan opened this issue · 7 comments

jingenyan commented

训练的图片如下图：
image
预测结果在conf>0.1时，根本检测不到任何目标

DataXujing commented

训练图片也只是用了1张？可否详细的说一下你训练集的情况？

jingenyan commented

为了测试模型是否会收敛，训练集：1张硬币的图像进行训练，并使用相同的图片进行测试，希望模型可以过拟合，可是发现最后map一直在0.05左右震荡，并且conf=0.1了，根本检测不到任何目标【正确或者错误的目标框】

DataXujing commented

我们有尝试过使用1张图像训练模型，个人观点： 训练集仅1张，模型无法再有限的训练集上学习更多的信息，同时我们也无法测试模型是否过拟合，这时的模型是学习不到好的参数的

jingenyan commented

@哦哦，你的意思是 使用一张图片进行训练模型很难收敛是吗，还是说，收敛了并且可以准确预测训练的那张图片，只是不能预测其他的测试图片，没有泛化能力

DataXujing commented

一张图图像训练出来的网络，大概率上也预测不出来这张图像上的object, 因为你的网络没有做判断的能力，不用纠结了，一张图就别在训练网络了，没有意义，这些模型是建立在大量数据的基础上的。没有任何意义

jingenyan commented

@DataXujing 好的，谢谢