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

权利要求书

1.一种基于PaddleDetection的混淆矩阵生成方法，其特征在于，包括以下步骤：

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

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

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

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

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

步骤六，创建excle表格，将步骤五得出的矩阵填充到表格中，绘制出用于评价模型实际应用效果的混淆矩阵。