少样本目标检测方法、装置和电子设备.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN114861842A(43)申请公布日2022.08.05(21)申请号202210797903.1(22)申请日2022.07.08(71)申请人中国科学院自动化研究所地址100190北京市海淀区中关村东路95号(72)发明人朱贵波李宗树葛国敬赵旭王金桥(74)专利代理机构北京路浩知识产权代理有限公司11002专利代理师赵娜(51)Int.Cl.G06K9/62(2022.01)权利要求书2页说明书10页附图3页(54)发明名称少样本目标检测方法、装置和电子设备(57)摘要本发明提供一种少样本目标检测方法、装置和电子设备；涉及机器视觉技术领域。该方法包括：采用包括基础类别的第一样本数据训练得到第一模型，第一模型包括支持分支、查询分支，第一模型的支持分支用于提取基础类别的类别特征；采用包括新类别的第二样本数据对第一模型进行微调，得到第二模型，第二模型的支持分支用于提取基础类别和新类别的类别特征，第二样本的数量远小于第一样本的数量；将待检测样本输入第二模型的查询分支，提取待检测样本的第一特征；将第一特征与第二模型的类别特征进行特征重加权处理，得到处理后的第二特征，利用第二特征确定待检测样本的检测结果。本发明能够提高少样本检测器的性能。CN114861842ACN114861842A权利要求书1/2页1.一种少样本目标检测方法，其特征在于，包括：采用包括基础类别的第一样本数据训练得到第一模型，所述第一模型包括支持分支、查询分支，所述第一模型的支持分支用于提取所述基础类别的类别特征；采用包括新类别的第二样本数据对所述第一模型进行微调，得到第二模型，所述第二模型的支持分支用于提取所述基础类别和所述新类别的类别特征，第二样本的数量远小于所述第一样本的数量；将待检测样本输入所述第二模型的查询分支，提取待检测样本的第一特征；将所述第一特征与所述第二模型的所述类别特征进行特征重加权处理，得到处理后的第二特征，利用所述第二特征确定所述待检测样本的检测结果。2.根据权利要求1所述的少样本目标检测方法，其特征在于，所述将所述第一特征与所述第二模型的所述类别特征进行特征重加权处理，得到处理后的第二特征，包括：将所述第二模型的所述类别特征作为卷积核，对所述第一特征进行卷积运算，得到第二特征，所述第二特征与所述第一特征的大小一致。3.根据权利要求2所述的少样本目标检测方法，其特征在于，所述将所述第二模型的所述类别特征作为卷积核，对所述第一特征进行卷积运算，得到第二特征，包括：将所述第二模型的所述类别特征作为第一卷积核对所述第一特征进行卷积运算，得到第一融合特征；将所述类别特征的第一个维度与第二个维度进行转置，将转置后的类别特征作为第二卷积核，采用第二卷积核对所述第一融合特征再次进行卷积运算；将卷积运算的结果与所述第一特征相加，得到第二特征；其中，所述类别特征的所述第一个维度为类别数，所述第二个维度为通道数。4.根据权利要求1所述的少样本目标检测方法，其特征在于，所述查询分支包括检测层；所述利用所述第二特征确定所述待检测样本的检测结果，包括：将所述第二特征输入所述检测层，获得所述待检测样本的目标类别，以及对应的目标边界框。5.根据权利要求1所述的少样本目标检测方法，其特征在于，所述查询分支还包括特征重加权网络，采用包括基础类别的第一样本数据训练得到第一模型，包括：将所述第一样本数据划分为支持集和查询集，所述支持集与所述查询集中均包括所述基础类别；将所述支持集输入所述支持分支，对每个所述基础类别提取类别特征；将所述查询集中的第一样本输入所述查询分支，提取所述第一样本的样本特征；将所述类别特征与所述样本特征输入所述特征重加权网络进行融合，利用融合后的特征确定所述第一样本的第一检测结果，以训练第一模型。6.根据权利要求5所述的少样本目标检测方法，其特征在于，所述训练第一模型，包括：通过损失函数计算所述第一检测结果与所述第一样本的标签之间的差异度，基于所述差异度使用反向传播算法调整所述第一模型的参数。7.一种少样本目标检测装置，其特征在于，包括：第一训练模块，用于采用包括基础类别的第一样本数据训练得到第一模型，所述第一模型包括支持分支、查询分支，所述第一模型的支持分支用于提取基础类别的类别特征；2CN114861842A权利要求书2/2页第二训练模块，用于采用包括基础类别和新类别的第二样本数据对所述第一模型进行微调，得到第二模型，所述第二模型的支持分支用于提取所述基础类别和所述新类别的类别特征，所述第二样本的数量远小于第一样本的数量；数据输入模块，用于将待检测样本输入所述第二模型的查询分支，提取所述待检测样本的第一特征；结果确定模块，用于将所述第一特征与所述第二模型的所述类别特征进行特征重加权处理，得