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")

YOLOX结果输出PR曲线代码相关推荐

python画pr曲线代码_利用Python中的numpy包实现PR曲线和ROC曲线的计算
闲来无事,边理解PR曲线和ROC曲线,边写了一下计算两个指标的代码.在 python 环境下,sklearn里有现成的函数计算ROC曲线坐标点,这里为了深入理解这两个指标,写代码的时候只用到numpy ...
python画pr曲线代码_Yolov3测试图及绘制PR曲线,yoloV3,map,和,画
训练指令:./darknet.exe detector train data/obj.data yolo-obj.cfg darknet53.conv.74 | tee train_yolov3.lo ...
【SLAM十四讲】ch11 回环检测词袋法实验得出相似分数后计算PR曲线 VPR实验编辑中
[SLAM十四讲]ch11 回环检测词袋法实验得出相似分数后计算PR曲线 [SLAM十四讲]ch11 回环检测词袋法实验得出相似分数后计算PR曲线 DBow3库安装 ch11编译 ch11 词 ...
机器学习笔记（一）P-R曲线与ROC曲线
P-R曲线与ROC曲线一.什么是P-R曲线? 二.什么是ROC曲线? 三.P-R曲线与ROC曲线有什么用? 三.绘制P-R曲线代码一.什么是P-R曲线? 要知道什么是P-R曲线,首先,我们要先了解 ...
PR曲线的概念和应用
目录一.PR曲线的概念二.精准率和召回率三.PR曲线的功能说明四.PR曲线代码实现一.PR曲线的概念 PR曲线中的P代表的是precision(精准率),R代表的是recall(召回率),其 ...
机器学习学习笔记(二)---PR曲线
学习笔记 PR曲线一.P和R代表什么 TP FP TN FN 基本概念 P与R 的定义评估PR曲线二.绘制PR曲线代码参考文献一.P和R代表什么在信息检索中,我们经常会关心"检 ...
多条Precision-Recall（PR）曲线绘制（PR曲线）含python代码
对于一个二分类问题,我们可以根据模型的分类结果将其分为四类:TP.FP.TN.FN. 查准率和查全率是一对矛盾的度量,一般来说,查准率高时,查全率往往偏低:而查全率高时,查准率往往偏低.例如,我们只要 ...
分类器MNIST交叉验证准确率、混淆矩阵、精度和召回率（PR曲线）、ROC曲线、多类别分类器、多标签分类、多输出分类
本博客是在Jupyter Notebook下进行的编译. 目录 MNIST 训练一个二分类器使用交叉验证测量精度混淆矩阵精度和召回率精度/召回率权衡 ROC曲线多类别分类器错误分析多标签 ...
Python多分类问题pr曲线绘制（含代码）
研究了三天的多分类pr曲线问题终于在昨天晚上凌晨一点绘制成功了!! 现将所学所感记录一下,一来怕自己会忘可以温故一下,二来希望能给同样有疑惑的铁子们一些启迪! 下图为我画的pr曲线,因为准确度超过了9 ...

YOLOX结果输出PR曲线代码

YOLOX结果输出PR曲线代码相关推荐

最新文章

热门文章