一种面向RD时频数据的深度学习模型评测系统及方法与流程

文档序号:27826609发布日期:2021-12-07 20:54阅读:299来源:国知局
一种面向RD时频数据的深度学习模型评测系统及方法与流程
一种面向rd时频数据的深度学习模型评测系统及方法
技术领域
1.本发明涉及航空航天、人工智能领域,特别涉及一种面向rd时频数据的深度学习模型评测系统及方法。


背景技术:

2.随着深度学习在图像目标检测等计算机视觉方向的不断应用与快速发展,深度学习在越来越多领域战胜了传统方法,特别是在小样本学习的推动下,人们有信心在有标注的数据量较小的前提下,使用深度学习得到优异的效果。rd时频数据是通过雷达等装置采集到得,经过时频分析和图像绘制可以得到具有物理意义的图像。在对rd时频数据的应用中,目标检测是最受瞩目的领域之一。传统的基于rd时频数据的的目标检测方法往往涉及较多的基于经验的分析和判断,需要根据具体数据表征采用相应的分析方法,方法较为庞杂。而面向rd时频数据使用深度学习模型进行目标检测是未来发展的重心,rd时频数据集的采集与构建比较困难,原因在于获取rd时频数据需要专业设备与装置,采集到得数据还需要降噪、时频分析和图像绘制等过程,而且相比很多目标检测的常见数据集,rd时频数据的目标较小、轮廓模糊,干扰物与真实目标更难以区分。因此可以应用于目标检测的rd时频数据集的体量较小,一般在1000到2000张左右。根据一般规律,将把数据集中20%到30%的样本作为验证集,然而实际应用中由于训练数据珍贵,验证集的数量会在此基础上下调,因此带来的验证集数量不足问题对模型评测结果影响很大。面向rd时频数据基于深度学习模型的实时目标检测是目前具有很大潜力的研究领域,因此需要从尽可能全面的角度衡量深度学习模型的能力,除了多维度上的准确性指标,还要考虑模型的时间复杂度。因此,提出一种面向rd时频数据的深度学习模型评测方法十分必要。
3.从rd时频数据灰度图像的数据特征出发,设计一种面向rd时频数据的基于图像融合与检验机制的验证集扩充方法,通过将rd时频灰度图像中的目标框剪出形成目标池,对目标进行图像处理后与背景进行随机融合,通过复制粘贴盲检测的融合结果则纳入验证集。设计一种面向rd时频数据的考虑多维度准确性与时间性能的深度模型评测方法,综合考虑多交并比(iou)阈值下的f1得分、多iou阈值下的各类别平均精度(map)、浮点运算次数(flops)三个指标,使用加权得分表示模型整体能力。因此,面向样本规模较小的rd时频数据,设计基于图像融合与检验机制的验证集扩充方法和考虑多维度准确性与时间性能的深度模型评测方法,可有效提高模型评测的准确性和有效性,更好地检验与衡量模型的整体能力。
4.目前国内外的图像增广方法较多,但是尚无一种面相rd时频数据图像的验证集扩充方法,国内外的单独采用多iou阈值下的f1得分、多iou阈值下的map、flops之一作为模型评测指标的较为常见,但是尚无采用加权得分表示模型整体效果的实例。因此,开发一种面向rd时频数据的深度学习模型评测方法具有重要意义和实用价值。


技术实现要素:

5.本发明解决的技术问题是:克服现有技术的不足,提供一种面向rd时频数据的深度学习模型评测系统及方法,包括验证集扩充方法及基于集成策略的模型评测方法,通过图像融合与检验机制对验证集进行扩充,可以在不引入明显噪声信息、不泄露目标标签信息、不改变数据分布情况下得到更大规模的验证集,使得验证集更好地表征整体数据的特征,更好地评测模型的泛化性能;用多iou阈值的f1得分、多iou阈值的map、flops的加权得分表示深度学习模型的整体能力,可以更准确地表征模型在单点最优、全局平均最优、时间性能方面的能力,从而为面向rd时频数据的深度学习模型评测提供有力的技术支撑。
6.本发明的技术方案是:一种面向rd时频数据的深度学习模型评测系统,包括验证集扩充模块、多iou阈值f1得分计算模块、多iou阈值map计算模块、flops计算模块和评测指标集成计算模块;
7.验证集扩充模块接收灰度化后的训练集和验证集、复制粘贴盲检测允许偏差、扩充数量作为输入,通过分析训练集和验证集整体数据中标签位置的整体分布,将训练集和验证集组成的整体数据图像中的目标剪出形成目标池,将背景挖空处理形成背景池,满足约束条件下进行目标图像处理并与背景随机融合,然后计算复制粘贴盲检测得分,根据得分是否满足得分要求,自动搜索图像处理参数或更换融合的目标与背景,直到满足得分要求则将融合图像纳入验证集,扩充过程完成后得到扩充后的验证集;
8.多iou阈值f1得分计算模块将模型在扩充后的验证集上的预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的召回率recall、精确率precision、f1得分,然后计算各个类别的平均f1得分,最后计算各个iou阈值下的平均f1得分;
9.多iou阈值map计算模块将模型在扩充后的验证集上的预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的召回率recall、精确率precision、平均精确率ap,计算各个类别ap的平均值map,计算各个iou阈值下map的平均值;
10.flops计算模块将外置的深度学习模型结构配置作为输入,自动分析模型中的卷积层、池化层、全连接层的个数,根据维度和法则自动计算flops;
11.评测指标集成计算模块对多iou阈值f1得分、多iou阈值map、flops进行量纲化,计算得到加权和作为深度学习模型评测得分。
12.所述训练集和验证集整体数据中标签位置的整体分布包括:每个图像中目标个数分布、目标左上角坐标分布、目标旋转角度分布、复制粘贴盲检测得分分布。
13.所述验证集扩充模块内部的具体工作流程为:
14.步骤11、提示用户录入灰度化的训练集和验证集、扩充数量、复制粘贴盲检测允许偏差,统计训练集和验证集整体数据的目标左上角坐标范围分布、目标旋转角度分布、复制粘贴盲检得分范围;
15.步骤12、在训练集和验证集组成的整体数据上,将每个图像上的目标框剪切出来组成目标池,背景处填充背景色组成背景池,记录数据集中已经存在的<目标,背景>的配对关系;
16.步骤13、判断验证集扩充数量是否达到用户输入的扩充数量,若未达到,跳转执行步骤15;
17.步骤14、在步骤13的判断中,若验证集扩充数量达到用户输入的扩充数量,则退出
验证集扩充模块工作流程;
18.步骤15、判断是否已经遍历完目标池和背景池,如果未遍历完,则跳转执行步骤17;
19.步骤16、在步骤15的判断中,如果已遍历完,则退出验证集扩充模块工作流程;
20.步骤17、随机从目标池和背景池中选择目标图像和背景图像,保证<目标,背景>配对关系没有出现过;
21.步骤18、对待融合的目标进行图像处理,处理手段包括背景透明和边缘模糊,背景透明参数和边缘模糊参数有预设的阈值范围和步长;
22.步骤19、根据整体数据的目标左上角坐标范围分布、目标旋转角度分布,生成随机融合参数:目标左上角坐标和旋转角度,将目标图像根据融合参数与背景进行融合;
23.步骤110、计算融合图像的复制粘贴盲检测得分,包括特征点提取、特征点匹配、聚类和几何评估,将复制粘贴盲检测得分量纲化到0到100范围内,分数越高表示图片是通过复制粘贴形式生成的概率越低;
24.步骤111、在原始数据的复制粘贴盲检得分范围基础上,范围下限向下偏移用户输入的允许偏差,判断当前融合图像的复制粘贴盲检测得分是否满足要求,如果满足要求则执行步骤112;
25.步骤112、将当前融合的图像加入验证集中,然后返回步骤13;这里只考虑每个图片中存在一个目标的情况,如果对于多目标情况,则需要将融合后的图片再次加入背景池并做好记录。
26.步骤113、在步骤111中,如果当前融合图像的复制粘贴盲检测得分不满足要求,则判断是否已经遍历完图像处理调参范围,如果已经遍历完,则返回步骤15;
27.步骤114、在步骤113的判断中,如果未遍历完图像处理调参范围,则调整图像处理参数,然后返回步骤18。
28.所述多iou阈值f1得分计算模块内部的具体工作流程为:
29.步骤21、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
30.步骤22、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的f1平均得分;
31.步骤23、在步骤22的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i,之后执行步骤24;
32.步骤24、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的f1平均得分;
33.步骤25、在步骤24的判断中,如果未遍历完全部分类,则获取下一个类别j,之后执行步骤26;
34.步骤26、判断是否已遍历完所有待评测验证集图片,如果已遍历完,则计算类别j的各个图像的f1得分的平均值;
35.步骤27、在步骤26的判断中,如果未遍历完所有待评测验证集图片,则获取下一个图像g,之后执行步骤28;
36.步骤28、对于步骤27获取到的图像,筛选出其中和当前类别j的第k个标注框的iou大于等于iou阈值i的若干个预测框;
37.步骤29、按照置信度降序顺序对步骤28筛选出的预测框进行排序;
38.步骤210、根据iou阈值i,构建混淆矩阵,计算正检数tp、漏检数fn、误检数fp;
39.步骤211、计算精度计算召回率
40.步骤212、计算f1得分计算完成后返回步骤26。
41.所述多iou阈值map计算模块内部的具体工作流程为:
42.步骤31、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
43.步骤32、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的map平均得分;
44.步骤33、在步骤32的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i,之后执行步骤34;
45.步骤34、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的ap平均得分;
46.步骤35、在步骤34的判断中,如果未遍历完全部分类,则获取下一个类别j,之后执行步骤36;
47.步骤36、根据预设的置信度取值范围,在置信度阈值取值范围内进行等距取值,获得n个置信度阈值;
48.步骤37、筛选出评测图像中置信度大于等于第m个置信度阈值的若干个预测框;
49.步骤38、将若干个预测框分别与若干个标注框进行匹配;
50.步骤39、根据iou阈值i,构建混淆矩阵,计算与第m个置信度阈值相对应的正检数tp、漏检数fn、误检数fp;
51.步骤310、计算与第m个置信度阈值相对应的精度计算召回率
52.步骤311、判断是否已遍历完所有置信度阈值,如果未遍历完,则返回步骤37;
53.步骤312、在步骤311的判断中,如果已遍历完,则根据各个置信度下的precision和recall绘制p

r曲线获得分类j的ap,然后返回步骤34。
54.所述flops计算模块内部的具体工作流程为:
55.步骤41、提示用户输入深度学习模型结构配置,并计算所有卷积层的flops,每个卷积层flops计算方式为:flops=[(c
i
×
k
w
×
k
h
)+(c
i
×
k
w
×
k
h

1)+1]
×
c0×
w
×
h,其中c
i
×
k
w
×
k
h
表示一次卷积操作中的乘法运算量,(c
i
×
k
w
×
k
h

1)+1]表示一次卷积操作中的加法运算量;
[0056]
步骤42、计算所有池化层的flops,每个池化层flops计算方式为flops=b
×
(h
×
w
×
c)
×
k2,其中b是批量大小,h
×
w是输出特征图的尺寸,c是输出通道数,k2是每个点乘法运算量;
[0057]
步骤43、计算所有全连接层的flops,每个全连接层flops计算方式为flops=[i+
(i

1)+1]=(2
×
i)
×
o,其中第一个i表示乘法运算量,(i

1)表示加法运算量;
[0058]
步骤44、将步骤41、42、43的flops求和,得到模型整体的flops,退出flops计算模块工作流程。
[0059]
所述评测指标集成计算模块的工作流程为:
[0060]
步骤51、读取多iou阈值f1得分计算结果;
[0061]
步骤52、读取多iou阈值map计算结果;
[0062]
步骤53、读取flops计算结果;
[0063]
步骤54、根据多iou阈值f1得分计算结果、多iou阈值map计算结果、flops计算结果的值域范围和特征,对三个结果进行同量纲化;
[0064]
步骤55、根据预设的多iou阈值f1得分计算结果权重、多iou阈值map计算结果权重、flops权重,计算三者的加权和,作为面向rd时频数据的深度学习模型评测指标。
[0065]
一种面向rd时频数据的深度学习模型评测方法,步骤如下:
[0066]
1)验证集扩充模块接收灰度化后的训练集和验证集、复制粘贴盲检测允许偏差、扩充数量作为输入,通过分析训练集和验证集整体数据中标签位置的整体分布,将训练集和验证集组成的整体数据图像中的目标剪出形成目标池,将背景挖空处理形成背景池,满足约束条件下进行目标图像处理并与背景随机融合,然后计算复制粘贴盲检测得分,根据得分是否满足得分要求,自动搜索图像处理参数或更换融合的目标与背景,直到满足得分要求则将融合图像纳入验证集;
[0067]
2)多iou阈值f1得分计算模块将模型在扩充后的验证集上的预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的召回率recall、精确率precision、f1得分,然后计算各个类别的平均f1得分,最后计算各个iou阈值下的平均f1得分;
[0068]
3)多iou阈值map计算模块将模型在扩充后的验证集上的预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的召回率recall、精确率precision、平均精确率ap,计算各个类别ap的平均值map,计算各个iou阈值下map的平均值;
[0069]
4)flops计算模块将外置的深度学习模型结构配置作为输入,自动分析模型中的卷积层、池化层、全连接层的个数,根据维度和法则自动计算flops;
[0070]
5)评测指标集成计算模块对多iou阈值f1得分、多iou阈值map、flops进行量纲化,计算得到加权和作为深度学习模型评测得分。
[0071]
所述训练集和验证集整体数据中标签位置的整体分布包括:每个图像中目标个数分布、目标左上角坐标分布、目标旋转角度分布、复制粘贴盲检测得分分布。
[0072]
所述验证集扩充模块内部的具体工作流程为:
[0073]
步骤11、提示用户录入灰度化的训练集和验证集、扩充数量、复制粘贴盲检测允许偏差,统计训练集和验证集整体数据的目标左上角坐标范围分布、目标旋转角度分布、复制粘贴盲检得分范围;
[0074]
步骤12、在训练集和验证集组成的整体数据上,将每个图像上的目标框剪切出来组成目标池,背景处填充背景色组成背景池,记录数据集中已经存在的<目标,背景>的配对关系;
[0075]
步骤13、判断验证集扩充数量是否达到用户输入的扩充数量,若未达到,跳转执行步骤15;
[0076]
步骤14、在步骤13的判断中,若验证集扩充数量达到用户输入的扩充数量,则退出验证集扩充模块工作流程;
[0077]
步骤15、判断是否已经遍历完目标池和背景池,如果未遍历完,则跳转执行步骤17;
[0078]
步骤16、在步骤15的判断中,如果已遍历完,则退出验证集扩充模块工作流程;
[0079]
步骤17、随机从目标池和背景池中选择目标图像和背景图像,保证<目标,背景>配对关系没有出现过;
[0080]
步骤18、对待融合的目标进行图像处理,处理手段包括背景透明和边缘模糊,背景透明参数和边缘模糊参数有预设的阈值范围和步长;
[0081]
步骤19、根据整体数据的目标左上角坐标范围分布、目标旋转角度分布,生成随机融合参数:目标左上角坐标和旋转角度,将目标图像根据融合参数与背景进行融合;
[0082]
步骤110、计算融合图像的复制粘贴盲检测得分,包括特征点提取、特征点匹配、聚类和几何评估,将复制粘贴盲检测得分量纲化到0到100范围内,分数越高表示图片是通过复制粘贴形式生成的概率越低;
[0083]
步骤111、在原始数据的复制粘贴盲检得分范围基础上,范围下限向下偏移用户输入的允许偏差,判断当前融合图像的复制粘贴盲检测得分是否满足要求,如果满足要求则执行步骤112;
[0084]
步骤112、将当前融合的图像加入验证集中,然后返回步骤13;这里只考虑每个图片中存在一个目标的情况,如果对于多目标情况,则需要将融合后的图片再次加入背景池并做好记录。
[0085]
步骤113、在步骤111中,如果当前融合图像的复制粘贴盲检测得分不满足要求,则判断是否已经遍历完图像处理调参范围,如果已经遍历完,则返回步骤15;
[0086]
步骤114、在步骤113的判断中,如果未遍历完图像处理调参范围,则调整图像处理参数,然后返回步骤18。
[0087]
所述多iou阈值f1得分计算模块内部的具体工作流程为:
[0088]
步骤21、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
[0089]
步骤22、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的f1平均得分;
[0090]
步骤23、在步骤22的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i,之后执行步骤24;
[0091]
步骤24、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的f1平均得分;
[0092]
步骤25、在步骤24的判断中,如果未遍历完全部分类,则获取下一个类别j,之后执行步骤26;
[0093]
步骤26、判断是否已遍历完所有待评测验证集图片,如果已遍历完,则计算类别j的各个图像的f1得分的平均值;
[0094]
步骤27、在步骤26的判断中,如果未遍历完所有待评测验证集图片,则获取下一个图像g,之后执行步骤28;
[0095]
步骤28、对于步骤27获取到的图像,筛选出其中和当前类别j的第k个标注框的iou
大于等于iou阈值i的若干个预测框;
[0096]
步骤29、按照置信度降序顺序对步骤28筛选出的预测框进行排序;
[0097]
步骤210、根据iou阈值i,构建混淆矩阵,计算正检数tp、漏检数fn、误检数fp;
[0098]
步骤211、计算精度计算召回率
[0099]
步骤212、计算f1得分计算完成后返回步骤26。
[0100]
所述多iou阈值map计算模块内部的具体工作流程为:
[0101]
步骤31、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
[0102]
步骤32、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的map平均得分;
[0103]
步骤33、在步骤32的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i,之后执行步骤34;
[0104]
步骤34、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的ap平均得分;
[0105]
步骤35、在步骤34的判断中,如果未遍历完全部分类,则获取下一个类别j,之后执行步骤36;
[0106]
步骤36、根据预设的置信度取值范围,在置信度阈值取值范围内进行等距取值,获得n个置信度阈值;
[0107]
步骤37、筛选出评测图像中置信度大于等于第m个置信度阈值的若干个预测框;
[0108]
步骤38、将若干个预测框分别与若干个标注框进行匹配;
[0109]
步骤39、根据iou阈值i,构建混淆矩阵,计算与第m个置信度阈值相对应的正检数tp、漏检数fn、误检数fp;
[0110]
步骤310、计算与第m个置信度阈值相对应的精度计算召回率
[0111]
步骤311、判断是否已遍历完所有置信度阈值,如果未遍历完,则返回步骤37;
[0112]
步骤312、在步骤311的判断中,如果已遍历完,则根据各个置信度下的precision和recall绘制p

r曲线获得分类j的ap,然后返回步骤34。
[0113]
所述flops计算模块内部的具体工作流程为:
[0114]
步骤41、提示用户输入深度学习模型结构配置,并计算所有卷积层的flops,每个卷积层flops计算方式为:flops=[(c
i
×
k
w
×
k
h
)+(c
i
×
k
w
×
k
h

1)+1]
×
c0×
w
×
h,其中c
i
×
k
w
×
k
h
表示一次卷积操作中的乘法运算量,(c
i
×
k
w
×
k
h

1)+1]表示一次卷积操作中的加法运算量;
[0115]
步骤42、计算所有池化层的flops,每个池化层flops计算方式为flops=b
×
(h
×
w
×
c)
×
k2,其中b是批量大小,h
×
w是输出特征图的尺寸,c是输出通道数,k2是每个点乘法运算量;
[0116]
步骤43、计算所有全连接层的flops,每个全连接层flops计算方式为flops=[i+(i

1)+1]=(2
×
i)
×
o,其中第一个i表示乘法运算量,(i

1)表示加法运算量;
[0117]
步骤44、将步骤41、42、43的flops求和,得到模型整体的flops,退出flops计算模块工作流程。
[0118]
所述评测指标集成计算模块的工作流程为:
[0119]
步骤51、读取多iou阈值f1得分计算结果;
[0120]
步骤52、读取多iou阈值map计算结果;
[0121]
步骤53、读取flops计算结果;
[0122]
步骤54、根据多iou阈值f1得分计算结果、多iou阈值map计算结果、flops计算结果的值域范围和特征,对三个结果进行同量纲化;
[0123]
步骤55、根据预设的多iou阈值f1得分计算结果权重、多iou阈值map计算结果权重、flops权重,计算三者的加权和,作为面向rd时频数据的深度学习模型评测指标。
[0124]
本发明与现有技术相比的优点在于:
[0125]
(1)通过本发明提供的适用于规模普遍较小的rd时频数据的验证集扩充方法及深度学习模型评测方法,可以在不引入明显噪声信息、不泄露目标标签信息、不改变数据分布情况得到更大规模的验证集,基于此采用融合多维度准确性和时间性能的加权指标作为深度学习模型评测的依据,可以更准确地反映模型的整体能力。
[0126]
(2)本发明设计了一种基于图像融合与检验机制的验证集扩充方法,通过复制粘贴盲检测对融合后的图像进行检验,该方法能够自动调节满足不引入明显噪声信息、不泄露目标标签信息条件的图像融合参数。该方法可以实现满足一定约束条件下的验证集自动化扩充,从而更好地评测深度学习模型的泛化性能。
[0127]
(3)本发明设计了一种基于集成策略的深度学习评测方法,通过综合考虑多iou阈值的f1得分、多iou阈值的map、flops三个指标,使用加权得分表示模型整体能力,可以更准确的表征模型在单点最优、全局平均最优、时间性能方面的能力。
[0128]
(4)通过本发明可显著提高面向rd时频数据的深度学习模型评测准确性,降低扩充数据所需的人力成本,更全面地表示深度学习模型在预测准确性与预测时间方面能力,为筛选出面向rd时频数据的具有实时目标检测潜力的深度学习模型提供有力的评测依据。因此,将该技术应用于工程实际具有重要意义。
附图说明
[0129]
图1为本发明系统结构图。
[0130]
图2为验证集扩充模块工作流程图。
[0131]
图3为多iou阈值f1得分计算模块工作流程图。
[0132]
图4为多iou阈值map计算模块工作流程图。
[0133]
图5为flops计算模块工作流程图。
[0134]
图6为评测指标集成计算模块工作流程图。
具体实施方式
[0135]
如图1

6所示,本发明一种面向rd时频数据的深度学习模型评测系统,由验证集扩
充模块、多iou阈值f1得分计算模块、多iou阈值map计算模块、flops计算模块、评测指标集成计算模块构成。
[0136]
所述的验证集扩充模块接收灰度化后的训练集和验证集、复制粘贴盲检测允许偏差、扩充数量作为输入,通过分析训练集和验证集整体数据中标签位置的整体分布,包括每个图像中目标个数分布、目标左上角坐标分布、目标旋转角度分布、复制粘贴盲检测得分分布,将图像中的目标剪出形成目标池,将背景挖空处理形成背景池,满足约束条件下进行目标图像处理并与背景随机融合,然后计算复制粘贴盲检测得分,根据得分是否满足得分要求,自动搜索图像处理参数或更换融合的目标与背景,直到满足得分要求则将融合图像纳入验证集。
[0137]
验证集扩充完成后,多iou阈值f1得分计算模块将验证集预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的recall、precision、f1得分,然后计算各个类别的平均f1得分,最后计算各个iou阈值下的平均f1得分。
[0138]
多iou阈值f1得分计算完成后,多iou阈值map计算模块将验证集预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的recall、precision、ap,计算各个类别ap的平均值得到map,计算各个iou阈值下map的平均值。
[0139]
多iou阈值map计算完成后,flops计算模块将深度学习模型结构配置作为输入,自动分析模型中的卷积层、池化层、全连接层的个数,根据维度和法则自动计算flops。
[0140]
flops计算完成后,对多iou阈值f1得分、多iou阈值map、flops进行量纲化,计算加权和作为深度学习模型评测得分。
[0141]
所述验证集扩充模块的工作流程为:
[0142]
1、提示用户录入灰度化的训练集和验证集、扩充数量、复制粘贴盲检测允许偏差,统计训练集和验证集整体数据的目标左上角坐标范围分布、目标旋转角度分布、复制粘贴盲检得分范围;
[0143]
2、在训练集和验证集组成的整体数据上,将每个图像上的目标框剪切出来组成目标池,背景处填充背景色组成背景池,记录数据集中已经存在的<目标,背景>的配对关系;
[0144]
3、判断验证集扩充数量是否达到用户输入的扩充数量,若未达到,继续执行步骤5;
[0145]
4、在步骤3的判断中,若验证集扩充数量达到用户输入的扩充数量,则退出验证集扩充模块;
[0146]
5、判断是否已经遍历完目标池和背景池,如果未遍历完,则继续执行步骤7;
[0147]
6、在步骤5的判断中,如果已遍历完,则退出验证集扩充模块;
[0148]
7、随机从目标池和背景池中选择目标图像和背景图像,保证<目标,背景>配对关系没有出现过;
[0149]
8、对待融合的目标进行图像处理,处理手段包括背景透明和边缘模糊,背景透明参数和边缘模糊参数有预设的阈值范围和步长;
[0150]
9、根据整体数据的目标左上角坐标范围分布、目标旋转角度分布,生成随机融合参数:目标左上角坐标和旋转角度,将目标图像根据融合参数与背景进行融合;
[0151]
10、计算融合图像的复制粘贴盲检测得分,一般包括特征点提取、特征点匹配、聚类和几何评估,将复制粘贴盲检测得分量纲化到0到100范围内,分数越高表示图片是通过
复制粘贴形式生成的概率越低。
[0152]
11、在原始数据的复制粘贴盲检得分范围基础上,范围下限向下偏移用户输入的允许偏差,判断当前融合图像的复制粘贴盲检测得分是否满足要求,如果满足要求则继续执行步骤12。
[0153]
12、将当前融合的图像加入验证集中,然后返回步骤3。这里只考虑每个图片中存在一个目标的情况,如果对于多目标情况,则需要将融合后的图片再次加入背景池并做好记录。
[0154]
13、在步骤11中,如果当前融合图像的复制粘贴盲检测得分不满足要求,则判断是否已经遍历完图像处理调参范围,如果已经遍历完,则返回步骤5。
[0155]
14、在步骤13的判断中,如果未遍历完图像处理调参范围,则调整图像处理参数,然后返回步骤8。
[0156]
所述多iou阈值f1得分计算模块的工作流程为:
[0157]
1、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
[0158]
2、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的f1平均得分;
[0159]
3、在步骤2的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i;
[0160]
4、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的f1平均得分;
[0161]
5、在步骤4的判断中,如果未遍历完全部分类,则获取下一个类别j;
[0162]
6、判断是否已遍历完所有待评测验证集图片,如果已遍历完,则计算类别j的各个图像的f1得分的平均值;
[0163]
7、在步骤6的判断中,如果未遍历完所有待评测验证集图片,则获取下一个图像g;
[0164]
8、对于步骤7获取到的图像,筛选出其中和当前类别j的第k个标注框的iou大于等于iou阈值i的若干个预测框;
[0165]
9、按照置信度降序顺序对步骤8筛选出的预测框进行排序;
[0166]
10、根据iou阈值i,构建混淆矩阵,计算正检数tp、漏检数fn、误检数fp;
[0167]
11、计算精度precision,计算召回率recall,
[0168]
12、计算f1得分,计算完成后返回步骤6。
[0169]
所述多iou阈值map计算模块的工作流程为:
[0170]
1、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
[0171]
2、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的map平均得分;
[0172]
3、在步骤2的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i;
[0173]
4、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的ap平均得分;
[0174]
5、在步骤4的判断中,如果未遍历完全部分类,则获取下一个类别j;
[0175]
6、根据预设的置信度取值范围,在置信度阈值取值范围内进行等距取值,获得n个置信度阈值;
[0176]
7、筛选出评测图像中置信度大于等于第m个置信度阈值的若干个预测框;
[0177]
8、将若干个预测框分别与若干个标注框进行匹配;
[0178]
9、根据iou阈值i,构建混淆矩阵,计算与第m个置信度阈值相对应的正检数tp、漏检数fn、误检数fp;
[0179]
10、计算与第m个置信度阈值相对应的精度precision,计算召回率recall,
[0180]
11、判断是否已遍历完所有置信度阈值,如果未遍历完,则返回步骤7;
[0181]
12、在步骤11的判断中,如果已遍历完,则根据各个置信度下的precision和recall绘制p

r曲线获得分类j的ap,然后返回步骤4。
[0182]
所述flops计算模块的工作流程为:
[0183]
1、提示用户输入深度学习模型结构配置,并计算所有卷积层的flops,每个卷积层flops计算方式为:flops=[(c
i
×
k
w
×
k
h
)+(c
i
×
k
w
×
k
h

1)+1]
×
c0×
w
×
h,其中c
i
×
k
w
×
k
h
表示一次卷积操作中的乘法运算量,(c
i
×
k
w
×
k
h

1)+1]表示一次卷积操作中的加法运算量;
[0184]
2、计算所有池化层的flops,每个池化层flops计算方式为flops=b
×
(h
×
w
×
c)
×
k2,其中b是批量大小,h
×
w是输出特征图的尺寸,c是输出通道数,k2是每个点乘法运算量;
[0185]
3、计算所有全连接层的flops,每个全连接层flops计算方式为flops=[i+(i

1)+1]=(2
×
i)
×
o,其中第一个i表示乘法运算量,(i

1)表示加法运算量;
[0186]
4、将步骤1、2、3的flops求和,得到模型整体的flops,退出flops计算模块;
[0187]
所述评测指标集成计算模块的工作流程为:
[0188]
1、读取多iou阈值f1得分计算结果;
[0189]
2、读取多iou阈值map计算结果;
[0190]
3、读取flops计算结果;
[0191]
4、根据多iou阈值f1得分计算结果、多iou阈值map计算结果、flops计算结果的值域范围和特征,对三个结果进行同量纲化;
[0192]
5、根据预设的多iou阈值f1得分计算结果权重、多iou阈值map计算结果权重、flops权重,计算三者的加权和,作为面向rd时频数据的深度学习模型评测指标。
[0193]
本发明还涉及一种面向rd时频数据的深度学习模型评测方法,步骤如下:
[0194]
1)验证集扩充模块接收灰度化后的训练集和验证集、复制粘贴盲检测允许偏差、扩充数量作为输入,通过分析训练集和验证集整体数据中标签位置的整体分布,将训练集和验证集组成的整体数据图像中的目标剪出形成目标池,将背景挖空处理形成背景池,满足约束条件下进行目标图像处理并与背景随机融合,然后计算复制粘贴盲检测得分,根据得分是否满足得分要求,自动搜索图像处理参数或更换融合的目标与背景,直到满足得分要求则将融合图像纳入验证集;
[0195]
2)多iou阈值f1得分计算模块将模型在扩充后的验证集上的预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的召回率recall、精确率precision、f1得分,然后计算各个类别的平均f1得分,最后计算各个iou阈值下的平均f1得分;
[0196]
3)多iou阈值map计算模块将模型在扩充后的验证集上的预测结果作为输入,在每个iou阈值下,计算验证集内每个类别的召回率recall、精确率precision、平均精确率ap,计算各个类别ap的平均值map,计算各个iou阈值下map的平均值;
[0197]
4)flops计算模块将外置的深度学习模型结构配置作为输入,自动分析模型中的卷积层、池化层、全连接层的个数,根据维度和法则自动计算flops;
[0198]
5)评测指标集成计算模块对多iou阈值f1得分、多iou阈值map、flops进行量纲化,计算得到加权和作为深度学习模型评测得分。
[0199]
所述训练集和验证集整体数据中标签位置的整体分布包括:每个图像中目标个数分布、目标左上角坐标分布、目标旋转角度分布、复制粘贴盲检测得分分布。
[0200]
所述验证集扩充模块内部的具体工作流程为:
[0201]
步骤11、提示用户录入灰度化的训练集和验证集、扩充数量、复制粘贴盲检测允许偏差,统计训练集和验证集整体数据的目标左上角坐标范围分布、目标旋转角度分布、复制粘贴盲检得分范围;
[0202]
步骤12、在训练集和验证集组成的整体数据上,将每个图像上的目标框剪切出来组成目标池,背景处填充背景色组成背景池,记录数据集中已经存在的<目标,背景>的配对关系;
[0203]
步骤13、判断验证集扩充数量是否达到用户输入的扩充数量,若未达到,跳转执行步骤15;
[0204]
步骤14、在步骤13的判断中,若验证集扩充数量达到用户输入的扩充数量,则退出验证集扩充模块工作流程;
[0205]
步骤15、判断是否已经遍历完目标池和背景池,如果未遍历完,则跳转执行步骤17;
[0206]
步骤16、在步骤15的判断中,如果已遍历完,则退出验证集扩充模块工作流程;
[0207]
步骤17、随机从目标池和背景池中选择目标图像和背景图像,保证<目标,背景>配对关系没有出现过;
[0208]
步骤18、对待融合的目标进行图像处理,处理手段包括背景透明和边缘模糊,背景透明参数和边缘模糊参数有预设的阈值范围和步长;
[0209]
步骤19、根据整体数据的目标左上角坐标范围分布、目标旋转角度分布,生成随机融合参数:目标左上角坐标和旋转角度,将目标图像根据融合参数与背景进行融合;
[0210]
步骤110、计算融合图像的复制粘贴盲检测得分,包括特征点提取、特征点匹配、聚类和几何评估,将复制粘贴盲检测得分量纲化到0到100范围内,分数越高表示图片是通过复制粘贴形式生成的概率越低;
[0211]
步骤111、在原始数据的复制粘贴盲检得分范围基础上,范围下限向下偏移用户输入的允许偏差,判断当前融合图像的复制粘贴盲检测得分是否满足要求,如果满足要求则执行步骤112;
[0212]
步骤112、将当前融合的图像加入验证集中,然后返回步骤13;这里只考虑每个图
片中存在一个目标的情况,如果对于多目标情况,则需要将融合后的图片再次加入背景池并做好记录。
[0213]
步骤113、在步骤111中,如果当前融合图像的复制粘贴盲检测得分不满足要求,则判断是否已经遍历完图像处理调参范围,如果已经遍历完,则返回步骤15;
[0214]
步骤114、在步骤113的判断中,如果未遍历完图像处理调参范围,则调整图像处理参数,然后返回步骤18。
[0215]
所述多iou阈值f1得分计算模块内部的具体工作流程为:
[0216]
步骤21、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
[0217]
步骤22、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的f1平均得分;
[0218]
步骤23、在步骤22的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i,之后执行步骤24;
[0219]
步骤24、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的f1平均得分;
[0220]
步骤25、在步骤24的判断中,如果未遍历完全部分类,则获取下一个类别j,之后执行步骤26;
[0221]
步骤26、判断是否已遍历完所有待评测验证集图片,如果已遍历完,则计算类别j的各个图像的f1得分的平均值;
[0222]
步骤27、在步骤26的判断中,如果未遍历完所有待评测验证集图片,则获取下一个图像g,之后执行步骤28;
[0223]
步骤28、对于步骤27获取到的图像,筛选出其中和当前类别j的第k个标注框的iou大于等于iou阈值i的若干个预测框;
[0224]
步骤29、按照置信度降序顺序对步骤28筛选出的预测框进行排序;
[0225]
步骤210、根据iou阈值i,构建混淆矩阵,计算正检数tp、漏检数fn、误检数fp;
[0226]
步骤211、计算精度计算召回率
[0227]
步骤212、计算f1得分计算完成后返回步骤26。
[0228]
所述多iou阈值map计算模块内部的具体工作流程为:
[0229]
步骤31、提示用户输入验证集预测结果,根据预设iou阈值范围和步长,得到iou阈值的所有值;
[0230]
步骤32、判断是否已遍历完iou阈值,如果已遍历完,则计算各iou的map平均得分;
[0231]
步骤33、在步骤32的判断中,如果未遍历完iou阈值,则获取下一个iou阈值i,之后执行步骤34;
[0232]
步骤34、判断是否已遍历完全部分类,如果已遍历完,则计算各类别的ap平均得分;
[0233]
步骤35、在步骤34的判断中,如果未遍历完全部分类,则获取下一个类别j,之后执行步骤36;
[0234]
步骤36、根据预设的置信度取值范围,在置信度阈值取值范围内进行等距取值,获得n个置信度阈值;
[0235]
步骤37、筛选出评测图像中置信度大于等于第m个置信度阈值的若干个预测框;
[0236]
步骤38、将若干个预测框分别与若干个标注框进行匹配;
[0237]
步骤39、根据iou阈值i,构建混淆矩阵,计算与第m个置信度阈值相对应的正检数tp、漏检数fn、误检数fp;
[0238]
步骤310、计算与第m个置信度阈值相对应的精度计算召回率
[0239]
步骤311、判断是否已遍历完所有置信度阈值,如果未遍历完,则返回步骤37;
[0240]
步骤312、在步骤311的判断中,如果已遍历完,则根据各个置信度下的precision和recall绘制p

r曲线获得分类j的ap,然后返回步骤34。
[0241]
所述flops计算模块内部的具体工作流程为:
[0242]
步骤41、提示用户输入深度学习模型结构配置,并计算所有卷积层的flops,每个卷积层flops计算方式为:flops=[(c
i
×
k
w
×
k
h
)+(c
i
×
k
w
×
k
h

1)+1]
×
c0×
w
×
h,其中c
i
×
k
w
×
k
h
表示一次卷积操作中的乘法运算量,(c
i
×
k
w
×
k
h

1)+1]表示一次卷积操作中的加法运算量;
[0243]
步骤42、计算所有池化层的flops,每个池化层flops计算方式为flops=b
×
(h
×
w
×
c)
×
k2,其中b是批量大小,h
×
w是输出特征图的尺寸,c是输出通道数,k2是每个点乘法运算量;
[0244]
步骤43、计算所有全连接层的flops,每个全连接层flops计算方式为flops=[i+(i

1)+1]=(2
×
i)
×
o,其中第一个i表示乘法运算量,(i

1)表示加法运算量;
[0245]
步骤44、将步骤41、42、43的flops求和,得到模型整体的flops,退出flops计算模块工作流程。
[0246]
所述评测指标集成计算模块的工作流程为:
[0247]
步骤51、读取多iou阈值f1得分计算结果;
[0248]
步骤52、读取多iou阈值map计算结果;
[0249]
步骤53、读取flops计算结果;
[0250]
步骤54、根据多iou阈值f1得分计算结果、多iou阈值map计算结果、flops计算结果的值域范围和特征,对三个结果进行同量纲化;
[0251]
步骤55、根据预设的多iou阈值f1得分计算结果权重、多iou阈值map计算结果权重、flops权重,计算三者的加权和,作为面向rd时频数据的深度学习模型评测指标。
[0252]
本发明虽然已以较佳实施例公开如上,但其并不是用来限定本发明,任何本领域技术人员在不脱离本发明的精神和范围内,都可以利用上述揭示的技术内容对本发明技术方案做出可能的变动和修改,因此,凡是未脱离本发明技术方案的内容,依据本发明的技术实质对以上实施例所作的任何简单修改、等同变化及修饰,均属于本发明技术方案的保护范围。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1