object-detection - 如何理解 yolov4 训练是否过拟合？

Question

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

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

在训练结束时，它给出了这个图表：

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

Answer 1

这不是过拟合，测试集上的准确率低于验证集上的准确率也就不足为奇了。该模型已经在通常更接近验证集的训练集上学习，并且有时模型在使用验证集时会进行微调，因此预计该模型将在验证集上表现更好（对于每个 ML 模型都是如此）。

如果训练集的准确率不断增加，而验证的准确率在 epoc 之间下降，则会在训练时发生过拟合。（过拟合指的是训练集）。