MMdetection如何保存最优的checkpoint文件

先来看用训练好的checkpoint文件跑test.py脚本的输出结果

最下方会输出一个有序字典OrderedDict，我们可以指定相关的键来保存最好的模型。

需要的操作就是在配置文件中进行如下修改：

evaluation = dict(interval=1, metric='bbox', save_best='auto')

save_best用于指定对应的键，'auto'是指保留第一个键对应最大值的checkpoint文件，即'bbox_mAP'（对应coco评价指标的第一行），也可以指定save_best='bbox_mAP_50'（coco评价指标的第二行）保留最大checkpoint文件。

保存的checkpoint文件路径在work_dir中，命名格式如下：

 这样在训练过程中会自动保存自己指定指标最高的checkpoint文件。