Detectron绘制loss曲线和pr曲线

绘制loss曲线

参见博客:detectron训练日志可视化

绘制p-r曲线

在detectron测试完毕后,会在相应的结果输出文件夹中输出每一类的不同precision对应的recall值,保存在xx_pr.pkl中,例如ship_pr.pkl,绘制p-r的代码为:

import pickle
import os.path as osp
import matplotlib.pyplot as plt

PR_path= ''
f = open(PR_path,'rb')  
info = pickle.load(f)
precision = info['prec']
recall = info['rec']
plt.plot(recall,precision)
plt.show()

猜你喜欢

转载自blog.csdn.net/Mr_health/article/details/84567873