一种基于迁移学习微调的DDETR小样本目标检测方法技术

技术编号:41595101 阅读:21 留言:0更新日期:2024-06-07 00:05
本发明专利技术提出了一种基于迁移学习微调的DDETR小样本目标检测方法。该方法提出小样本目标检测基础框架FS_DDETR,在原DDETR网络的基础上添加了类别过滤模块、类别重映射模块、分类结果映射模块和分类头部可调节的DDETR检测网络,实现了对于小样本目标检测器不同阶段、不同评估方法的支持;在该框架下设计了一种基于DDETR网络的迁移学习微调策略,使得FS_DDETR在相关数据集上的小样本目标检测任务中,达到了相当有竞争力的性能。

【技术实现步骤摘要】

本专利技术涉及一种基于迁移学习微调的ddetr小样本目标检测方法,属于计算机视觉。


技术介绍

1、目标检测是众多视觉任务,例如实例分割、目标跟踪、图像描述、行为识别等的基础,在视频监控、自动驾驶和机器人视觉导航等现实场景中也已经被应用。作为计算机视觉中最基础、也最具有挑战性的工作之一,目标检测近年来受到了广泛的关注,并取得了许多实质性进展。归功于大数据和计算机硬件的快速发展,基于深度学习的目标检测算法占据了主导地位。例如,2015年提出的fast r-cnn使用深度卷积网络对目标区域进行检测(rossgirshick.fast r-cnn.cvpr 2015),2020年提出的基于transformer的端到端目标检测网络detection transformer(detr,carion n,massa f,synnaeve g,et al.end-to-endobject detection with transformers.european conference on computer vision,2020:213-229),以及随后提出的defo本文档来自技高网...

【技术保护点】

1.一种基于迁移学习微调的DDETR小样本目标检测方法,其特征在于:

2.如权利要求1所述的以DDETR目标检测网络为基本网络框架实现基于迁移学习微调的小样本目标检测,其特征在于包括如下步骤:

3.如权利要求2所述的一种基于迁移学习微调的DDETR小样本目标检测方法,其特征在于步骤(2.4)中DDETR检测网络模型的训练步骤如下:

4.如权利要求2所述的一种基于迁移学习微调的DDETR小样本目标检测方法,其特征在于步骤(2.4)中基于DDETR网络的迁移学习微调策略包括:

【技术特征摘要】

1.一种基于迁移学习微调的ddetr小样本目标检测方法,其特征在于:

2.如权利要求1所述的以ddetr目标检测网络为基本网络框架实现基于迁移学习微调的小样本目标检测,其特征在于包括如下步骤:

3.如权利要求2所述的一种基于迁移学习...

【专利技术属性】
技术研发人员:郑锦张莉彬李航
申请(专利权)人:北京航空航天大学
类型:发明
国别省市:

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

1