yolov5 混淆矩阵计算精确率(percision)
·
yolov5 混淆矩阵计算精确率(percision)
刚才看到一个误人子弟的文章,发现还不少人评论收藏
yolov5的混淆矩阵
yolov5 输出的混淆矩阵是归一化之后的 计算召回率的话直接取对角线上的值就好了,但是想要计算精确率(percision))并不能直接算
首先我们要得到一个没有归一化的混淆矩阵:
找到 metrics.py 文件 然后找到 plot 中的normalize 改成 false 这样会输出未归一化的混淆矩阵

然后就可以计算精确率了
如第一个类别’'error’的
P of error = 25/(25+1) = 0.96 (精确率)
R of error = 25/(25+3) = 0.89 (召回率)
这个结果和终端执行val.py之后的结果也能对应的上
如果按照那个大聪明的说法
P of error = 0.89/(0.89+0.2) = 0.75 (精确率)

关于为什么终端和混淆矩阵输出的结果不一样这位作者说的挺清楚的
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐


所有评论(0)