mmdetection训练模型之后,可用以下函数对模型进行评价
一、日志分析
1、tools/analyze_logs.py 利用给定的训练日志文件,可以打印出loss函数与map的曲线,运行pip install serborn来安装对应的依赖。
analyze_logs.py 可以绘制loss, loss_cls, loss_bbox, acc,mAP等等
- 打印分类损失
python tools/analyze_logs.py plot_curve log.json --keys loss_cls --legend loss_cls
- 打印分类与回归的损失,并保存为一个pdf文件
python tools/analyze_logs.py plot_curve log.json --keys loss_cls loss_bbox --out losses.pdf
- 在同一个图中对比两个runs的map
python tools/analyze_logs.py plot_curve log1.json log2.json --keys bbox_mAP --legend run1 run2
- 打印acc
python tools/analyze_logs.py plot_curve