代码如下所示:
import _pickle as cPickle
import matplotlib.pyplot as plt
fr1= open('people_pr.pkl', 'rb') # 这里open中第一个参数需要修改成自己生产的pkl文件
inf1 = cPickle.load(fr1)
fr1.close()
x1= inf1['rec']
y1= inf1['prec']
fr2= open('people_pr_win.pkl', 'rb') # 这里open中第一个参数需要修改成自己生产的pkl文件
inf2 = cPickle.load(fr2)
fr2.close()
x2= inf2['rec']
y2= inf2['prec']
plt.figure()
plt.xlabel('recall')
plt.ylabel('precision')
plt.title('PR cruve')
plt.plot(x1, y1, color="red")
plt.plot(x2, y2, color="blue")
plt.show()
#print('AP:', inf['ap'])
pkl文件获取步骤 参考博客:https://blog.csdn.net/qq_33350808/article/details/83178002注 意
pkl文件应该使用下面输出文件名
里面的pkl 而不是annots.pkl
python reval_voc_py3.py --voc_dir <voc文件路径> --year <年份> --image_set <验证集文件名> --classes <类名文件路径><输出文件夹名>