detectron2 结果可视化

时间:2024-04-05 22:05:34

Detectron2 结果可视化

    Detectron是构建在Caffe2和Python之上计算机视觉库,集成了多项计算机视觉最新成果,一经发布广受好评。Facebook AI研究院又开源了Detectron的升级版,也就是Detectron2。参考了网上许多关于Detectron2输出结果可视化的程序,但是程序是对应于Detectron的,无法处理Detectron2的输出结果。
    博主改写了一个程序,可以顺利进行Detectron2结果可视化:
detectron2 结果可视化

    d2_loss_ visualization.pyDetectron2各类损失total_loss\loss_cls\loss_box_reg以及分类准确度cls_accuracy的可视化、曲线绘制。
  输入文件为Detectron2训练结束后的,根目录下生成的output/metrics.json文件
    代码地址:loss_curve
   好用请给代码点star