栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

YOLOX结果输出PR曲线代码

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

YOLOX结果输出PR曲线代码

      在网上搜了很多都没有找到相关代码,自己东西拼凑,没想到最后成功了。

      在yolox/data/voc.py中加入这几行代码,首先要保证算法能够正常跑通,训练完毕和eval后会在当前目录下自动生成PR曲线图,代码比较简短,可根据代码自行修改。

#放在voc.py在方法_do_python_eval中,for循环之前
classes_rec = []
classes_prec = []
classes_label = []

#放在上面方法for循环后面
classes_label.append(cls)
classes_prec.append(prec)
classes_rec.append(rec)

#放在if iou=0.5代码段最后面
pl.title('Precision-Recall Curve')
pl.xlabel("Recall")
pl.ylabel("Precision")
plt.grid(True)
pl.ylim([0.0, 1.05])
pl.xlim([0.0, 1.0])
color = ['r', 'g', 'b', 'c', 'm', 'y', 'k']
for i in range(len(classes_label)):
   pl.plot(classes_rec[i], classes_prec[i], label=classes_label[i], color=color[i])
pl.legend(loc="lower left")
plt.savefig("PR.jpg")

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1000824.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号