[发明专利]小样本目标检测模型构建方法、装置、设备及存储介质

说明书

本申请涉及一种小样本目标检测模型构建方法、装置、设备及存储介质，涉及计算机视觉图像目标检测技术领域，包括获取基类数据集和新类数据集；基于基类训练集对预设的神经网络模型进行训练生成基础目标检测模型；根据基类数据集和新类数据集构建平衡训练集；基于平衡训练集对基础目标检测模型进行训练生成小样本目标检测模型；其中，神经网络模型包括背景衰减块和幻觉器，背景衰减块用于对图像进行前景加强和背景抑制的处理，幻觉器用于根据基类的共享类内特征变化为新类生成特征级训练实例，其基于特征级训练实例能够从新类别物体中提取出新的特征。本申请通过引入背景衰减块和幻觉器来提高模型小样本检测的准确度。

技术领域

本申请涉及计算机视觉图像目标检测技术领域，特别涉及一种小样本目标检测模型构建方法、装置、设备及存储介质。

背景技术

目前基于深度学习的目标检测模型需要大量的标注样本进行训练，但在实际场景当中，往往很难获取到相应规模的高质量标注样本。而小样本学习可以在少量标签样本条件下实现对新类型目标的决速分类识别，降低了对大规模标签数据的依赖。受小样本学习启发，近年来，小样本图像目标检测技术研究逐渐引起广泛关注。相对于常规的监督式目标检测方法需要借助大星的待检测类别标注样本进行训练，小样本图像目标检测在具有充足标注数据的公开数据集基础上，只需要极少量待检测的新类别标注样本，通过精心设计训练方法、模型结构和损失函数，就可以获得具有一定泛化性能的检测摸型，极大提高模型开发的效率，也更具实用价值。由此可见，小样本图像目标检测是传统目标检测技术与小样本学习技术的融合，旨在通过少量的标注样本学习具有较好泛化性能的检测模型。

但是，由于传统的小样本目标检测模型对前景关注度不够且训练数据中缺乏变化，以致传统模型的检测准确度较低。

发明内容

本申请提供一种小样本目标检测模型构建方法、装置、设备及存储介质，以解决相关技术中由于传统的小样本目标检测模型对前景关注度不够且训练数据中缺乏变化而导致的检测准确度较低的问题。

第一方面，提供了一种小样本目标检测模型构建方法，包括以下步骤：

获取基类数据集和新类数据集；

基于所述基类训练集对预设的神经网络模型进行训练生成基础目标检测模型；

根据所述基类数据集和所述新类数据集构建平衡训练集；

基于所述平衡训练集对所述基础目标检测模型进行训练生成小样本目标检测模型；

其中，所述神经网络模型包括背景衰减块和幻觉器，所述背景衰减块用于对图像进行前景加强和背景抑制的处理，所述幻觉器用于根据基类的共享类内特征变化为新类生成特征级训练实例，所述幻觉器基于所述特征级训练实例能够从新类别物体中提取出新的特征。

一些实施例中，所述神经网络模型还包括骨干网络、区域建议网络RPN、池化层和预测层；

所述骨干网络用于对图像进行特征提取，得到初始特征图；

所述背景衰减块用于对初始特征图进行前景加强和背景抑制处理，得到注意力加强后的特征图；

所述RPN用于对注意力加强后的特征图进行锚框预测，得到多个推荐框；

所述池化层用于对注意力加强后的特征图和候选框进行池化处理，得到区域特征图；

所述预测层包括幻觉器、分类器和回归网络，所述幻觉器用于对区域特征图进行目标识别，得到目标对象；所述分类器用于对区域特征图和目标对象进行分类，得到分类结果；所述回归网络用于对区域特征图中的锚框进行边框回归处理，以修正锚框的位置。

一些实施例中，所述背景衰减块用于对初始特征图进行前景加强和背景抑制处理，得到注意力加强后的特征图，包括：