我正在训练以在自定义数据集中使用 YOLOv4 进行人体检测。我用这个命令来训练数据集:
!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg yolov4.conv.137 -dont_show -map
在训练结束时,它给出了这个图表:
验证最多可提供 97% 的准确率。但是当我观察测试数据时,它在视频录制中的准确度约为 80%。是否过拟合?我怎么解决这个问题?我认为图表中的准确性应该越来越高。
答案 0 :(得分:0)
这不是过拟合,测试集上的准确率低于验证集并不奇怪。 模型在训练集上学习,通常更接近验证集,有时模型在使用验证集时会被微调,因此预计模型在验证集上表现更好(对于每个 ML 模型都是如此) ).
如果训练集的准确度不断提高而验证的准确度在 epocs 之间下降,则在训练时会发生过度拟合。 (过拟合指的是训练集)