基于图像数据增强的图像训练集生成方法和模型训练方法技术

本发明专利技术提供了一种基于图像数据增强的图像训练集生成方法和模型训练方法，其中，图像训练集生成方法包括：构造初始图像训练集后训练初始目标检测模型；对原始训练样本图像进行重检测后输出检测输出图像；比较各检测输出图像上形成的检测框和原始训练样本图像上的标注框，将检测框框选的错误目标图像块裁切下；在选中的原始训练样本图像上裁切下剔除图像块，选取错误目标图像块补偿在剔除图像块的原始位置处形成负样本图像；混合进入初始图像训练集中形成图像训练集。本发明专利技术解决了利用现有技术中的数据增强方法扩充的图样训练集，能够提升训练出的目标检测模型的泛化能力，但无法提高目标检测模型对特定的目标对象的识别准确率的问题。确率的问题。确率的问题。

【技术实现步骤摘要】
基于图像数据增强的图像训练集生成方法和模型训练方法


[0001]本专利技术涉及图像处理以及模型训练
，具体而言，涉及一种基于图像数据增强的 图像训练集生成方法和模型训练方法。

技术介绍

[0002]现有的计算机视觉领域中，通常使用基于深度学习的目标检测模型来进行决策，为了达 到对目标检测模型理想的训练效果，在模型训练过程中需要使用到包含有大量图像样本的图 像训练集，而在图像样本在数量不足的情况下，便需要采用数据增强的手段来扩充图像训练 集。
[0003]相关技术中，通常采用几何变换、颜色变化或混合样本数据增强生成难例的方式来获得 更多的图像样本数据，上述的几种数据增强方法虽然能够有效地扩充图样训练集，丰富图样 训练集中的图像样本数据多样性，提升训练出的目标检测模型的泛化能力；但对于提升目标 检测模型针对特定的目标对象的识别准确率则没有明显帮助，从而导致现有的目标检测模型 存在一定的使用局限性。例如，在道路交通场景中，对道路上行驶车辆的车牌或驾乘人员的 准确监控识别，直接影响到道路交通的综治管理的稳定性、道路交通运输的安全性，因此，本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于图像数据增强的图像训练集生成方法，其特征在于，包括：步骤S1，采集多张目标场景图像后使用标注框框选出其中的目标对象，以形成多张原始训练样本图像，汇集多张所述原始训练样本图像构成初始图像训练集，并基于深度神经网络利用所述初始图像训练集训练形成初始目标检测模型；步骤S2，使用所述初始目标检测模型对所述初始图像训练集中的多张所述原始训练样本图像进行重检测，以形成一一对应的多张检测输出图像；比较各所述检测输出图像上形成的检测框和与其相对应的所述原始训练样本图像上的标注框的属性信息，以识别出所述检测输出图像上错误检测为所述目标对象的部分所述检测框，并将该部分所述检测框框选的错误目标图像块裁切下汇集形成负样本图像增强数据集；步骤S3，选取所述初始图像训练集中的至少一部分所述原始训练样本图像，在选中的每张所述原始训练样本图像上裁切下至少一个剔除图像块，针对裁切下的每个所述剔除图像块，使用所述负样本图像增强数据集中随机选取的一个所述错误目标图像块补偿在所述剔除图像块的原始位置处，以形成负样本图像；步骤S4，将生成的所有所述负样本图像混合进入所述初始图像训练集中形成图像训练集，所述图像训练集用于对所述初始目标检测模型优化训练。2.根据权利要求1所述的图像训练集生成方法，其特征在于，在所述步骤S3中，选取所述初始图像训练集中的一部分所述原始训练样本图像，在选中的每张所述原始训练样本图像上裁切下多个所述剔除图像块，使用多个所述错误目标图像块一一对应地补偿在多个所述剔除图像块的原始位置处，以形成负样本图像；在所述步骤S4中，将生成的所有所述负样本图像与所述初始图像训练集中的另一部分所述原始训练样本图像混合后形成所述图像训练集。3.根据权利要求2所述的图像训练集生成方法，其...

【专利技术属性】
技术研发人员：陈映，曹松，任必为，
申请(专利权)人：北京文安智能技术股份有限公司，
类型：发明
国别省市：