Detectron2 结果可视化
Detectron是构建在Caffe2和Python之上计算机视觉库,集成了多项计算机视觉最新成果,一经发布广受好评。Facebook AI研究院又开源了Detectron的升级版,也就是Detectron2。参考了网上许多关于Detectron2输出结果可视化的程序,但是程序是对应于Detectron的,无法处理Detectron2的输出结果。
博主改写了一个程序,可以顺利进行Detectron2结果可视化:
d2_loss_ visualization.py:Detectron2各类损失total_loss\loss_cls\loss_box_reg以及分类准确度cls_accuracy的可视化、曲线绘制。
输入文件为Detectron2训练结束后的,根目录下生成的output/metrics.json文件
代码地址:loss_curve
好用请给代码点star