利索能及
我要发布
收藏
专利号: 2021112759662
申请人: 杭州海康威视数字技术股份有限公司
专利类型:发明专利
专利状态:已下证
更新日期:2026-08-04
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种数据增广策略选择方法,其特征在于,所述方法包括:

获取多个数据增广策略各自对应的第一检测模型;每一数据增广策略对应的第一检测模型是基于该数据增广策略对应的第一数据集训练得到的,每一数据增广策略对应的第一数据集是:基于该数据增广策略对第二数据集进行数据增广处理得到的;

针对每一数据增广策略,利用该数据增广策略对应的第一检测模型,对第三数据集中第三样本图像中的目标物进行检测,得到各检测框的置信度,其中,所述第三数据集中不包括每一第三样本图像中目标物所对应的标识信息;

根据该数据增广策略所对应的检测框的数量和/或检测框的置信度,计算该数据增广策略对应的目标物检测性能值;

基于每一数据增广策略对应的目标物检测性能值,选择目标数据增广策略;

所述多个数据增广策略的获取方式包括:

基于候选库中包括的多个数据增广算法,选取多组种子策略,每组种子策略中包括第一数量个数据增广算法;

针对每组种子策略,基于第二数量,以及该组种子策略中各数据增广算法所对应随机概率区间/幅值变换区间的最大值和最小值,确定该组种子策略中各数据增广算法所对应的随机概率值/幅值,得到多个数据增广策略。

2.根据权利要求1所述的方法,其特征在于,所述获取多个数据增广策略各自对应的第一检测模型的步骤,包括:获取所述第二数据集;

获取多个数据增广策略;

针对每一数据增广策略,利用该数据增广策略对所述第二数据集进行数据增广处理,得到每一数据增广策略对应的第一数据集;

针对每一数据增广策略,利用该数据增广策略对应的第一数据集,对预设检测模型进行训练,得到每一数据增广策略所对应的第一检测模型。

3.根据权利要求1所述的方法,其特征在于,针对任一数据增广策略对应的第一数据集,采用以下步骤训练得到该数据增广策略对应的第一检测模型:获取所述第一数据集;所述第一数据集包括多个第一样本图像,以及每一第一样本图像中目标物的第一标识信息;

利用预设检测模型,对每一第一样本图像中的目标物进行检测,得到第一检测结果;

基于所述第一检测结果以及所述第一标识信息,计算所述预设检测模型的第一损失值;

若第一差值大于第一预设阈值,则调整所述预设检测模型的参数,并返回执行所述利用预设检测模型,对每一第一样本图像中的目标物进行检测,得到第一检测结果的步骤;所述第一差值为本轮训练得到的第一损失值与上一轮训练得到第一损失值间的差值;

若所述第一差值不大于所述第一预设阈值,则将当前的预设检测模型确定为训练好的第一检测模型。

4.根据权利要求3所述的方法,其特征在于,所述第一标识信息包括指示目标物所在区域的区域标识,以及指示目标物的类别的类别标识;

所述基于所述第一检测结果以及所述第一标识信息,计算所述预设检测模型的第一损失值的步骤,包括:计算所述第一检测结果中检测框所在的区域,与所述第一数据集中区域标识所指示的目标物所在的区域间的第一误差值;

计算所述第一检测结果中检测框所对应目标物的类别,与所述第一数据集中类别标识所指示目标物的类别间的第二误差值;

计算所述第一误差值和所述第二误差值和值,作为所述预设检测模型的第一损失值。

5.根据权利要求1所述的方法,其特征在于,所述根据该数据增广策略所对应的检测框的数量和/或检测框的置信度,计算该数据增广策略对应的目标物检测性能值的步骤,包括:计算该数据增广策略所对应的检测框的数量的和值,作为该数据增广策略对应的目标物检测性能值;和/或计算该数据增广策略所对应的检测框的置信度的和值,作为该数据增广策略对应的目标物检测性能值。

6.根据权利要求1所述的方法,其特征在于,所述基于每一数据增广策略对应的目标物检测性能值,选择目标数据增广策略的步骤,包括:将所述多个数据增广策略中所述目标物检测性能值最高的数据增广策略确定为目标数据增广策略。

7.根据权利要求1所述的方法,其特征在于,所述方法还包括:

获取第四数据集和第五数据集;其中,所述第四数据集包括多个第四样本图像,以及每一第四样本图像中目标物所对应的第三标识信息,所述第五数据集包括多个第五样本图像;

利用所述目标数据增广策略,对所述第四数据集进行数据增广处理,得到第六数据集;

利用所述目标数据增广策略对应的第一检测模型,对所述第六数据集中的每一第六样本图像中的目标物进行检测,得到第二检测结果;

利用所述目标数据增广策略对应的第一检测模型,对所述第五数据集中的每一第五样本图像中的目标物进行检测,得到第三检测结果;

基于所述第二检测结果和所述第三检测结果,计算所述目标数据增广策略对应的第一检测模型的第二损失值;

若第二差值大于第二预设阈值,则对所述目标数据增广策略对应的第一检测模型的参数进行微调,并返回执行所述利用所述目标数据增广策略对应的第一检测模型,对所述第六数据集中的每一第六样本图像中的目标物进行检测,得到第二检测结果的步骤;所述第二差值为本轮训练得到的第二损失值与上一轮训练得到第二损失值间的差值;

若所述第二差值不大于所述第二预设阈值,则将当前的第一目标物检测模型确定为训练好的第二检测模型。

8.根据权利要求7所述的方法,其特征在于,所述方法还包括:

获取待检测图像;

利用所述第二检测模型,对所述待检测图像进行目标物检测,得到第四检测结果。

9.一种数据增广策略选择装置,其特征在于,所述装置包括:

第一获取模块,用于获取多个数据增广策略各自对应的第一检测模型;每一数据增广策略对应的第一检测模型是基于该数据增广策略对应的第一数据集训练得到的,每一数据增广策略对应的第一数据集是:基于该数据增广策略对第二数据集进行数据增广处理得到的;

第一检测模块,用于针对每一数据增广策略,利用该数据增广策略对应的第一检测模型,对第三数据集中第三样本图像中的目标物进行检测,得到各检测框的置信度,其中,所述第三数据集中不包括每一第三样本图像中目标物所对应的标识信息;

第一计算模块,用于根据该数据增广策略所对应的检测框的数量和/或检测框的置信度,计算该数据增广策略对应的目标物检测性能值;

选择模块,用于基于每一数据增广策略对应的目标物检测性能值,选择目标数据增广策略;

所述第一获取模块,具体用于基于候选库中包括的多个数据增广算法,选取多组种子策略,每组种子策略中包括第一数量个数据增广算法;针对每组种子策略,基于第二数量,以及该组种子策略中各数据增广算法所对应随机概率区间/幅值变换区间的最大值和最小值,确定该组种子策略中各数据增广算法所对应的随机概率值/幅值,得到多个数据增广策略。

10.一种数据增广策略选择系统,其特征在于,所述系统包括图像采集设备和模型训练设备;

所述图像采集设备,用于获取多个第二样本图像以及每一第二样本图像中目标物所对应的标识信息,得到第二数据集,并获取多个第三样本图像作为第三数据集,其中,所述第三数据集中不包括每一第三样本图像中目标物所对应的标识信息;

所述模型训练设备,用于基于多个数据增广策略,分别对所述第二数据集进行数据增广处理,得到每一数据增广策略对应的第一数据集;基于每一数据增广策略对应的第一数据集训练得到该数据增广策略所对应的第一检测模型;

所述模型训练设备,还用于针对每一数据增广策略,利用该数据增广策略对应的第一检测模型,对所述第三数据集中第三样本图像中的目标物进行检测,得到各检测框的置信度;根据该数据增广策略所对应的检测框的数量和/或检测框的置信度,计算该数据增广策略对应的目标物检测性能值;基于每一数据增广策略对应的目标物检测性能值,选择目标数据增广策略;

所述模型训练设备,具体用于基于候选库中包括的多个数据增广算法,选取多组种子策略,每组种子策略中包括第一数量个数据增广算法;针对每组种子策略,基于第二数量,以及该组种子策略中各数据增广算法所对应随机概率区间/幅值变换区间的最大值和最小值,确定该组种子策略中各数据增广算法所对应的随机概率值/幅值,得到多个数据增广策略。

11.根据权利要求10所述的系统,其特征在于,所述系统还包括客户端;

所述图像采集设备,还用于在获得所述目标数据增广策略后,获取待检测图像;

所述客户端,用于利用所述目标数据增广策略对应的第一检测模型,对所述待检测图像进行目标物检测,得到第五检测结果。