一种基于自蒸馏数据增强的无人艇小样本目标检测方法技术

技术编号:41732176 阅读:14 留言:0更新日期:2024-06-19 12:53
本发明专利技术涉及一种基于自蒸馏数据增强的无人艇小样本目标检测方法,包括:获取待检测的目标图像,将其输入预训练好的目标检测模型,得到检测结果;该目标检测模型的训练过程包括:获取训练图像样本;对小样本进行数据增强得到增强样本,构建平衡数据集;对目标检测模型进行初步训练;利用平衡数据集对学生网络进行训练,将其参数传递给教师网络;利用扩充训练集对教师网络进行训练,生成伪标签样本;计算学生网络输出结果与教师网络输出结果的损失,根据所述损失反向传播更新学生网络的参数,循环迭代,当损失函数收敛时停止迭代,得到训练好的目标检测网络。采用本发明专利技术可以提高目标检测模型在小样本情况下的准确率和目标检测模型泛化能力。

【技术实现步骤摘要】

本专利技术属于目标检测,具体涉及一种基于自蒸馏数据增强的无人艇小样本目标检测方法


技术介绍

1、目标检测一直都是计算机视觉中的一个重要任务。随着深度学习的不断发展,基于深度卷积网络(deep convolutional neural networks,dcnn)的目标检测模型性能得到大幅提高,成为目前该领域的主流方法。基于深度卷积网络的目标检测模型大致可以分为两类:无锚的单阶段方法和基于锚的两阶段方法。

2、参照图1所示,单阶段的方法是以yolo和ssd为代表的一系列方法,此类方法通常采用密集的锚框或默认框来覆盖图像的不同位置和尺度,并通过分类和回归头部来预测每个框内是否存在目标以及目标的位置。这种方法简化了目标检测流程,提高了检测速度,但在处理小目标和遮挡等复杂场景时可能存在一定的挑战。

3、两阶段的方法主要包括rcnn系列模型,如r-cnn、fast-rcnn、fasterrcnn以及maskr-cnn等。这类模型首先会运用选择性搜索或类似的区域提取技术,从图像中筛选出潜在的候选区域。随后,对每个候选区域进行特征提取和分类处理本文档来自技高网...

【技术保护点】

1.一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,包括:

2.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,所述目标检测模型为基于卷积神经网络的目标检测模型,所述基于卷积神经网络的目标检测模型使用层次化视觉自注意力层作为主干网络。

3.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,所述基于卷积神经网络的目标检测模型包括多粒度特征模块,提取目标图像或训练样本更细粒度的特征。

4.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,在步骤B...

【技术特征摘要】

1.一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,包括:

2.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,所述目标检测模型为基于卷积神经网络的目标检测模型,所述基于卷积神经网络的目标检测模型使用层次化视觉自注意力层作为主干网络。

3.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,所述基于卷积神经网络的目标检测模型包括多粒度特征模块,提取目标图像或训练样本更细粒度的特征。

4.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,在步骤b中,对小样本进行数据增强的方式包括:水平翻转、裁剪、马赛克、拼接、旋转、平移、添加噪声和模糊锐化处理。

5.根据权利要求1所述的一种基于自蒸馏数据增强的无人艇小样本目标检测方法,其特征在于,在步骤b中,在初始的平衡数据集中,所述增强样本和所述基类图像样本的比例为1:1,平衡数据集还包括由教师网络训练得到的伪标签样本,所述伪标签样本与基类图像样本的比例为1:1。

6.根据权利要求1所述的一种基于自蒸馏数据...

【专利技术属性】
技术研发人员:王国胤石岩刘群
申请(专利权)人:重庆邮电大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1