我如何理解 yolov4 训练是否过拟合?

时间:2021-04-12 08:53:26

标签: object-detection yolo darknet yolov4 overfitting-underfitting

我正在训练以在自定义数据集中使用 YOLOv4 进行人体检测。我用这个命令来训练数据集:

!./darknet detector train data/obj.data cfg/custom-yolov4-detector.cfg yolov4.conv.137 -dont_show -map

在训练结束时,它给出了这个图表:

enter image description here

验证最多可提供 97% 的准确率。但是当我观察测试数据时,它在视频录制中的准确度约为 80%。是否过拟合?我怎么解决这个问题?我认为图表中的准确性应该越来越高。

1 个答案:

答案 0 :(得分:0)

这不是过拟合,测试集上的准确率低于验证集并不奇怪。 模型在训练集上学习,通常更接近验证集,有时模型在使用验证集时会被微调,因此预计模型在验证集上表现更好(对于每个 ML 模型都是如此) ).

如果训练集的准确度不断提高而验证的准确度在 epocs 之间下降,则在训练时会发生过度拟合。 (过拟合指的是训练集)

相关问题