【mmdetection】绘制训练结果中验证集测试的mAP曲线图

mAP可视化工具

在训练模型的时候，我们通常会在每个 epoch 训练完成后，对验证集做一下测试，看一下模型的效果如何。那么如何绘制每个 epoch 验证的 mAP 结果呢？

绘制工具：analyze_logs.py

mmdetection 中自带可视化分析工具，在 tools/analysis_tools 文件夹下，我们可用以下命令绘制每次验证结果的 mAP 结果，从而选择合适的模型进行后续测试。

python tools/analysis_tools/analyze_logs.py plot_curve ./work_dir/20201228_234809.log.json --keys mAP --legend mAP --out mAP.jpg

其中，可选则不同的 keys （如 loss 等），根据训练的日志文件绘制训练的验证结果。

结果展示

图中，一个点代表一个 epoch 的验证结果。例图中是 100 个 epoch 的模型训练验证结果。以此，可以选择合适的 epoch 对应的模型作为自己后续的测试模型。
是不是很简单，如果对你有帮助的话，点个赞吧！阿里嘎多！
（PS：儿童节 + 上海解封，今天是个好日子！）