[发明专利]一种基于PaddleDetection的混淆矩阵生成方法

说明书

本发明提供了一种基于PaddleDetection的混淆矩阵生成方法，包括：在PaddlePaddle平台下运用PaddleDetection目标检测开发套件训练目标检测网络；获取有标注的测试数据集，分为两个文件夹，一个文件夹放json文件，另一个放对应的图片文件，并得到真实的目标物体外接矩形框；运用infer.py图片文件进行预测，得到预测标签；得出预测修正得到的矩形检测框；将真实矩形框与预测修正得到的矩形检测框进行循环匹配，将统计得到的值记录在矩阵上；创建excle表格，将记录的矩阵填充到excle表格中，绘制出用于评价模型实际应用效果的混淆矩阵，本发明方法能通过目标检测网络预测的检测框与实际的目标检测框进行匹配，绘制在混淆矩阵上，以便对于模型各个种类的识别结果进行统计研究，优化网络模型的训练方法。

技术领域

本发明涉及深度学习目标检测领域，特别是指一种基于PaddleDetection的混淆矩阵生成方法。

背景技术

飞桨(PaddlePaddle)是集深度学习核心框架、工具组件和服务平台为一体的百度自主研发的开源深度学习平台。PaddleDetection是飞桨深度学习平台下的优秀的目标检测开发套件，提供多种主流目标检测、实例分割、关键点检测算法，并且将各个网络组件进行模块化、提供数据增强策略、损失函数策略等，模型的压缩和跨平台的的高性能部署能够帮助工业项目更好的完成落地。

对于目标检测而言，检测识别的准确率很大程度上取决于模型本身的识别精度，因此前期在对神经网络识别模型进行训练的时候需要找到一个合适的适合实际生产的评价指标。在PaddleDetection套件中评价神经网络模型评价指标是通过mAP来进行评价，不能反应实际分拣中的各个类别的识别精度。

混淆矩阵(Confusion Matrix)也称误差矩阵，用n行n列的矩阵来表示。在人工智能图像分类精度的评价中，主要用于比较分类结果和实际测得的值。混淆矩阵的每一列代表了预测类别，每一列的总数表示预测为该类别的数据的数目；每一行代表了数据的真实归属类别，每一行的数据总数表示该类别的数据实例的数目。每一列中的数值表示真实数据被预测为该类的数目.如图1所示，第一行第一列中的96表示有96个实际归属第一类的实例被预测为第一类，同理，第一行第二列的2表示有2个实际归属为第一类的实例被错误预测为第二类，依次类推。

发明内容

本发明的主要目的在于克服现有技术中的上述缺陷，提出一种基于PaddleDetection的混淆矩阵生成方法，能通过目标检测网络预测的检测框与实际的目标检测框进行匹配，将匹配的实际结果记录下来，绘制在混淆矩阵上，以便对于模型各个种类的识别结果进行统计研究，优化网络模型的训练方法。

本发明采用如下技术方案：

步骤一，在PaddlePaddle平台下运用PaddleDetection目标检测开发套件训练一个目标检测网络。

步骤二，准备一批新的有标注的测试数据集，分为两个文件夹，一个文件夹放json文件，另一个放对应的图片文件。运用openCV的rect函数对json文件下的轮廓标签进行处理可以得到真实的目标物体外接矩形框；

步骤三，运用PaddleDetection开发套件下的infer.py对步骤二中的图片文件进行预测，得到预测标签；

步骤四，以图片文件的数量i来循环遍历，利用模型对每张图片进行预测，得到图片中每个物体的检测框，对物体的检测框进行iou判断，对于同类且iou大于阈值的检测框删除其他的框，仅保留置信度得分最高的检测框、对于不同类的且iou大于阈值的，只保留一个检测框，其他的删除，并把相应的标签改为others，由此便可以得到预测修正后的矩形检测框；

步骤五，将步骤二得出的真实的目标物体外接矩形框与步骤四预测修正得到的矩形检测框进行循环匹配，将统计得到的值记录在矩阵上；