yolo 绘制PR曲线

时间:2024-04-03 17:47:41

yolo 绘制PR曲线
代码如下所示:



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 <类名文件路径><输出文件夹名>