一种基于PaddleDetection的混淆矩阵生成方法.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN114120057A(43)申请公布日2022.03.01(21)申请号202111320510.3G06N3/04(2006.01)(22)申请日2021.11.09G06N3/08(2006.01)(71)申请人华侨大学地址362000福建省泉州市丰泽区城东城华北路269号(72)发明人陈伟鑫杨建红房怀英林柏宏杨天成李建涛计天晨陈强杨宇轩谭国亿(74)专利代理机构厦门市首创君合专利事务所有限公司35204代理人连耀忠王婷婷(51)Int.Cl.G06V10/774(2022.01)G06K9/62(2022.01)G06V10/82(2022.01)权利要求书1页说明书3页附图2页(54)发明名称一种基于PaddleDetection的混淆矩阵生成方法(57)摘要本发明提供了一种基于PaddleDetection的混淆矩阵生成方法，包括：在PaddlePaddle平台下运用PaddleDetection目标检测开发套件训练目标检测网络；获取有标注的测试数据集，分为两个文件夹，一个文件夹放json文件，另一个放对应的图片文件，并得到真实的目标物体外接矩形框；运用infer.py图片文件进行预测，得到预测标签；得出预测修正得到的矩形检测框；将真实矩形框与预测修正得到的矩形检测框进行循环匹配，将统计得到的值记录在矩阵上；创建excle表格，将记录的矩阵填充到excle表格中，绘制出用于评价模型实际应用效果的混淆矩阵，本发明方法能通过目标检测网络预测的检测框与实际的目标检测框进行匹配，绘制在混淆矩阵上，以便对于模型各个种类的识别结果进行统计研究，优化网络模型的训练方法。CN114120057ACN114120057A权利要求书1/1页1.一种基于PaddleDetection的混淆矩阵生成方法，其特征在于，包括以下步骤：步骤一，在PaddlePaddle平台下运用PaddleDetection目标检测开发套件训练目标检测网络；步骤二，获取有标注的测试数据集，分为两个文件夹，一个文件夹存放json文件，另一个存放对应的图片文件；运用openCV的rect函数对json文件下的轮廓标签进行处理可以得到真实的目标物体外接矩形框；步骤三，运用PaddleDetection开发套件下的infer.py对步骤二中的图片文件进行预测，得到预测标签；步骤四，以图片文件的数量i来循环遍历，利用模型对每张图片进行预测，得到图片中每个物体的矩形检测框，对物体的检测框进行iou判断，对于同类且iou大于阈值的检测框，保留置信度得分最高的检测框；对于不同类的且iou大于阈值的，仅保留一个检测框，并把相应的标签改为others，得到预测修正后的矩形检测框；步骤五，将步骤二得出的真实的目标物体外接矩形框矩形框与步骤四预测修正后的矩形检测框进行循环匹配，将统计得到的值记录在矩阵上；步骤六，创建excle表格，将步骤五得出的矩阵填充到表格中，绘制出用于评价模型实际应用效果的混淆矩阵。2CN114120057A说明书1/3页一种基于PaddleDetection的混淆矩阵生成方法技术领域[0001]本发明涉及深度学习目标检测领域，特别是指一种基于PaddleDetection的混淆矩阵生成方法。背景技术[0002]飞桨(PaddlePaddle)是集深度学习核心框架、工具组件和服务平台为一体的百度自主研发的开源深度学习平台。PaddleDetection是飞桨深度学习平台下的优秀的目标检测开发套件，提供多种主流目标检测、实例分割、关键点检测算法，并且将各个网络组件进行模块化、提供数据增强策略、损失函数策略等，模型的压缩和跨平台的的高性能部署能够帮助工业项目更好的完成落地。[0003]对于目标检测而言，检测识别的准确率很大程度上取决于模型本身的识别精度，因此前期在对神经网络识别模型进行训练的时候需要找到一个合适的适合实际生产的评价指标。在PaddleDetection套件中评价神经网络模型评价指标是通过mAP来进行评价，不能反应实际分拣中的各个类别的识别精度。[0004]混淆矩阵(ConfusionMatrix)也称误差矩阵，用n行n列的矩阵来表示。在人工智能图像分类精度的评价中，主要用于比较分类结果和实际测得的值。混淆矩阵的每一列代表了预测类别，每一列的总数表示预测为该类别的数据的数目；每一行代表了数据的真实归属类别，每一行的数据总数表示该类别的数据实例的数目。每一列中的数值表示真实数据被预测为该类的数目.如图1所示，第一行第一列中的96表示有96个实际归属第一类的实例被预测为第一类，同理，第一行第二列的2表示有2个实际归属为第一类的实例被错误预测为第二类，依次类推。发明内容[0005]本