mAP计算目标检测精确度实现源码
好多人都想算一下目标检测的精确度，mAP的概念虽然不好理解，但是理解了就很懂。
GITHUB代码下载
1、IOU的概念
IOU的概念应该比较简单，就是衡量预测框和真实框的重合程度。下图是一个示例：图中绿色框为实际框（好像不是很绿……），红色框为预测框，当我们需要判断两个框之间的关系时，主要就是判断两个框的重合程度。
计算IOU的公式为：
可以看到IOU是一个比值，即交并比。
在分子部分，值为预测框和实际框之间的重叠区域；
在分母部分，值为预测框和实际框所占有的总区域。
交区域和并区域的比值，就是IOU。
2、TP TN FP FN的概念
TP TN FP FN里面一共出现了4个字母，分别是T F P N。
T是True；
F是False；
P是Positive；
N是Negative。
T或者F代表的是该样本 是否被正确分类。
P或者N代表的是该样本 被预测成了正样本还是负样本。
TP（True Positives）意思就是被分为了正样本，而且分对了。
TN（True Negatives）意思就是被分为了负样本，而且分对了，
FP（False Positives）意思就是被分为了正样本，但是分错了（事实上这个样本是负样本）。
FN（False Negatives）意思就是被分为了负样本，但是分错了（事实上这个样本是这样本）。
在mAP计算的过程中主要用到了，TP、FP、FN这三个概念。
3、precision（精确度）和recall（召回率）
TP是分类器认为是正样本而且确实是正样本的例子，FP是分类器认为是正样本但实际上不是正样本的例子，Precision翻译成中文就是“分类器认为是正类并且确实是正类的部分占所有分类器认为是正类的比例”。
TP是分类器认为是正样本而且确实是正样本的例子，FN是分类器认为是负样本但实际上不是负样本的例子，Recall翻译成中文就是“分类器认为是正类并且确实是正类的部分占所有确实是正类的比例”。
4、概念举例
如图所示，蓝色的框是 真实框。绿色和红色的框是 预测框，绿色的框是正样本，红色的框是负样本。一般来讲，当预测框和真实框IOU>=0.5时，被认为是正样本。
因此对于这幅图来讲。
真实框一共有3个，正样本一共有2个，负样本一共有2个。
此时
5、单个指标的局限性
在目标检测算法里面有一个非常重要的概念是置信度，如果置信度设置的高的话，预测的结果和实际情况就很符合，如果置信度低的话，就会有很多误检测。
假设一幅图里面总共有3个正样本，目标检测对这幅图的预测结果有10个，其中3个实际上是正样本，7个实际上是负样本。对应置信度如下。
如果我们将可以接受的置信度设置为0.95的话，那么目标检测算法就会将序号为1的样本作为正样本，其它的都是负样本。此时TP = 1，FP = 0，FN = 2。
此时Precision非常高，但是事实上我们只检测出一个正样本，还有两个没有检测出来，因此只用Precision就不合适。
这个时候如果我们将可以接受的置信度设置为0.35的话，那么目标检测算法就会将序号为1的样本作为正样本，其它的都是负样本。此时TP = 3，FP = 3，FN = 0。
此时Recall非常高，但是事实上目标检测算法认为是正样本的样本里面，有3个样本确实是正样本，但有三个是负样本，存在非常严重的误检测，因此只用Recall就不合适。
二者进行结合才是评价的正确方法。
什么是AP
AP事实上指的是，利用不同的Precision和Recall的点的组合，画出来的曲线下面的面积。如下面这幅图所示。
当我们取不同的置信度，可以获得不同的Precision和不同的Recall，当我们取得置信度够密集的时候，就可以获得非常多的Precision和Recall。
此时Precision和Recall可以在图片上画出一条线，这条线下部分的面积就是某个类的AP值。
mAP就是所有的类的AP值求平均。
绘制mAP
我们首先在下载绘制mAP所需的代码。
在这个代码中，如果想要绘制mAP则需要三个内容。分别是：
detection-results：指的是预测结果的txt。
ground-truth：指的是真实框的txt。
image-optional：指的是图片，有这个可以可视化，但是这个可以没有。
我们需要生成这三个内容，此时下载第二个库，这个是我拿我制作的ssd代码写的一个可以生成对应txt的例子。
我们首先将整个VOC的数据集放到VOCdevikit中
然后修改voc2ssd.py里面的trainval_percent，一般用数据集的10%或者更少用于测试。如果大家放进VOCdevikit的数据集不是全部数据，而是已经筛选好的测试数据集的话，那么就把trainval_percent设置成0，表示全部的数据都用于测试。
然后运行voc2ssd.py。此时会生成test.txt，存放用于测试的图片的名字。
然后依次运行主目录下的get_dr_txt.py和get_gt_txt.py获得预测框对应的txt和真实框对应的txt。get_dr_txt.py是用来检测测试集里面的图片的，然后会生成每一个图片的检测结果，我重写了detect_image代码，用于生成预测框的txt。利用for循环检测所有的图片。
get_dr_txt.py是用来获取测试集中的xml，然后根据每个xml的结果生成真实框的txt。利用for循环检测所有的xml。
完成后我们会在input获得三个文件夹。
此时把input内部的文件夹复制到mAP的代码中的input文件夹内部就可以了，然后我们运行mAP的代码中的main.py，运行结束后，会生成mAP相关的文件。
结果生成在Result里面。
以上就是mAP计算目标检测精确度实现源码的详细内容，更多关于mAP目标检测的资料请关注脚本之家其它相关文章！
相关文章
在计算机视觉领域,经常需要对图像进行各种几何变换,如旋转、缩放和平移等,本文主要介绍了如何使用OpenCV实现图像的透视变换,需要的可以参考下2024-11-11
神经网络可能会让人感到恐惧，特别是对于新手机器学习的人来说。这篇文章主要介绍了基于Tensorflow搭建一个神经网络的实现，从入门开始，感兴趣的可以了解一下2021-05-05
邮件合并是现代办公中一项显著提升效率的技术,它巧妙地将大量个体数据与预设的文档模板相结合,实现了一次性批量生成定制化文档,下面我们就来看看如何使用Python实现在Word文档中进行邮件合并吧2024-04-04
在命令行中直接使用Python标准库的模块，最大的好处就是就是不用写代码，就能使用其中的功能，当临时需要一些某些功能的时候，用这种方式会快捷，方便很多，这篇文章主要介绍了Python标准库中隐藏的利器,需要的朋友可以参考下2023-11-11
今天小编就为大家分享一篇Python绘图实现显示中文，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
大家好，本篇文章主要讲的是Python中ROC曲线绘制，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2021-12-12
Tkinter是Python标准库中内置的图形用户界面（GUI）工具包,提供了创建窗口、按钮、文本框等GUI元素的功能,本文将介绍Tkinter的基础知识,帮助大家快速入门2023-12-12
这篇文章主要介绍了Python中property函数用法,结合实例形式分析了property函数的功能、参数、使用方法及相关操作注意事项,需要的朋友可以参考下2018-06-06
为何人工智能(AI)首选Python？读完这篇文章你就知道了（推荐）
这篇文章主要介绍了为何人工智能(AI)首选Python，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-04-04
python+Django+pycharm+mysql 搭建首个web项目详解
这篇文章主要介绍了python+Django+pycharm+mysql 搭建首个web项目,结合实例形式详细分析了python+Django+pycharm+mysql搭建web项目的具体步骤与相关操作技巧,需要的朋友可以参考下2019-11-11
最新评论