YOLO模型结果中的混淆矩阵怎么看

news/2024/7/11 1:02:52 标签: YOLO, python, 机器学习

大家在训练完YOLOv5、v8模型时,输出的结果中会有如图1所示。

confusion matrix.png

图1 训练结果中的混淆矩阵

那么这个图应该怎么看呢,请看下面的讲解。希望对你有帮助。

一、混淆矩阵

首先,先介绍一下什么是混淆矩阵。

混淆矩阵是一种将算法性能可视化的表现形式,矩阵的每一行代表一个真实的类别,每一列代表的是预测的类别。其实由上图就可以看出,横坐标是True,纵坐标是Precision。

由图中的每一格数字就能算出TP、FP、FN和TN,然后可以衍生出其他更多的评价指标,如P、R、F1等。

二、TP、FP与FN的回顾

目标检测中的混淆矩阵与分类非常相近,但是区别就在于分类任务是一张张图片,而检测任务是包含定位与分类两个任务,并且检测的对象是图像中目标类别。

图2 类别的判别类型

    目前最常用的判断一个检测结果是否正确的方式就是计算检测框与真实框的IOU,然后根据IOU去判别两个框是否匹配。以图2为例,红色的为模型预测的结果,绿色为真实标注框,两个框的IOU大于阈值,就被判定为匹配,同时两个框对应的类别也相同,那么此为正确的检测结果(TP)。

图2中的第二张图虽然IOU大于了阈值,但是由于类别不正确,所以被判定为误检(FP)。第三张图的检测框IOU小于阈值,没有与真实的标注框匹配,所以被判定为背景的误检。第四张图没有检测框,因此属于漏检(FN)

三、目标检测中的混淆矩阵

图3 

以类别pedestrian为例。在图3中,绿框(行)与红框(纵)相交的格子为pedestrian正确检测(TP)的概率0.51。而0.51右侧除了background(背景)外的所有格子都是误检(FP)的概率,根据横坐标可以看出都分别被误检成什么类别,比如pedestrian有0.02的概率被误判成bicycle类别。

从图3中可以知道pedestrian类别有0.22的概率被漏检(FN)。

根据例子就可以得出其他类别的TP、FN及FP的情况啦~


-------------------------------今天不学习,明天变废物!------------------------------------


http://www.niftyadmin.cn/n/5250295.html

相关文章

Android蓝牙协议栈fluoride(三) - 系统管理

关键线程 在整个协议栈中一共分三个主要线程:bt_jni_workqueue、bt_workqueue/btu message loop、hci_thread。从名称可以看出它们分别处理着各层的事务: bt_jni_workqueue:处理bt interface层的事务bt_workqueue/btu message loop&#xf…

三种入耳检测光感芯片驱动开发比较

三种入耳检测光感芯片驱动开发比较 是否需要申请加入数字音频系统研究开发交流答疑群(课题组)?可加我微信hezkz17, 本群提供音频技术答疑服务,群赠送语音信号处理降噪算法,蓝牙耳机音频,DSP音频项目核心开发资料, 重要的寄存器…

STM32(PWM、ADC)

1、PWM 定义 PWM,全称为脉冲宽度调制(Pulse Width Modulation),它通过改变信号的高电平和低电平的持续时间比例来控制输出信号的平均功率或电压。 PWM,全称为脉冲宽度调制(Pulse Width Modulation&#xff…

代码随想录二刷 | 二叉树 |填充每个节点的下一个右侧节点指针

代码随想录二刷 | 二叉树 |填充每个节点的下一个右侧节点指针 题目描述解题思路代码实现 题目描述 116.填充每个节点的下一个右侧节点指针 给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义…

百面嵌入式专栏(岗位分析)海康高级linux开发工程师分析

文章目录 一、岗位的介绍二、刨析2.1、掌握调试工具2.2、块设备相关知识 三、简历建议 沉淀、分享、成长,让自己和他人都能有所收获!😄 📢本篇我们将对海康高级linux开发工程师岗位进行分析 。 一、岗位的介绍 地点:上…

纸质签到太麻烦?快学习扫码签到的方法吧

以前参加会议或者活动时,经常会需要通过在前台的纸质文件中登记参会信息,当人员比较多时,需要长时间的排队等待,效率低而且不方便。现在很多人都会生成一个二维码图片打印出来,让来访人员通过扫描二维码来登记自己的信…

如何在Java中实现多条件排序

文章目录 前言一、在Java中使用Comparator接口实现多条件排序二、视频讲解总结 前言 多条件排序是一个复杂而实用的技术问题,它涉及到算法、数据结构等多个领域。 在实际工作中,我们常常需要根据多个条件对数据进行排序,以达到最优化的效果。…

vue2 echarts饼状图,柱状图,折线图,简单封装以及使用

vue2 echarts饼状图,柱状图,折线图,简单封装以及使用 1. 直接上代码(复制可直接用,请根据自己的文件修改引用地址,图表只是简单封装,可根据自身功能,进行进一步配置。) …