Tensorflow对象检测-mAP稳定,但总损耗正在下降。继续训练吗?

时间:2019-06-24 02:11:58

标签: python tensorflow object-detection tensorboard object-detection-api

使用Tensorflow 1.12.2和Python 3.6训练Object Detection分类器。使用4个类别中的大约20,000张图像,并且在经过16万步之后,mAP相当稳定,但是总损耗仍在下降(尽管非常缓慢)。我应该继续训练直到总损失停止下降或上升吗?

1 个答案:

答案 0 :(得分:0)

当“总损失”上升时,很可能是您过度拟合了。

一个好主意是对照测试数据集检查当前快照,看看是否获得满意的结果。如果结果良好,我将停止训练。