绘制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()