基于改进Faster R-CNN的小样本目标检测算法OA
A Few-Shot Object Detection Algorithm Based on Improved Faster R-CNN
针对小样本场景下训练样本数量稀少导致的模型可学习的信息有限、模型的性能表现受目标的尺度变化制约以及对基类知识的灾难性遗忘等问题,遵循迁移学习方法,提出了一种改进Faster R-CNN的小样本目标检测算法.首先,在特征提取阶段,加入中心坐标注意力机制,引导模型学习目标的核心特征,同时引入多尺度融合与可变形卷积,使模型能够在不同尺度特征图上提取丰富的特征信息;然后,引入渐进式RPN,采用两阶段回归方法,对生成的区域建议进行逐步优化;最后,引入知识补偿模块,借助解耦机制将先验知识与新知识进行解耦,保持对基类的良好记忆.实验结果表明,在PASCAL VOC数据集的3种新类划分中,所提算法与主流小样本目标检测方法FSODCR相比,在2-Shot和5-Shot设置下的nAP分别平均提升了8.37百分点和6.63百分点;在MS COCO数据集中,所提算法较FSODCR,在10-Shot下nAP和nAP75分别提升了1.2百分点和0.9百分点;在30-Shot下nAP和nAP75分别提升了1.2百分点和0.8百分点,显著提升了小样本条件下目标检测的精度与稳定性.
In order to solve the problems of limited information that can be learned by the model due to the scarcity of training samples in few-shot scenarios,the performance of the model is restricted by the scale change of the object,and the catastrophic forgetting of the base class knowledge,a few-shot object detection algorithm with improved Faster R-CNN was proposed by following the transfer learning method.Firstly,in the feature extraction stage,the central coordinate attention mechanism was added to guide the model to learn the core features of the object,and multi-scale fusion and deformable convolution were introduced to enable the model to extract rich feature information on the feature map at different scales.Then,a gradual RPN was introduced,and a two-stage regression method was used to gradually optimize the generated regional suggestions.Finally,the knowledge compensation module was introduced to decouple the prior knowledge from the new knowledge with the help of the decoupling mechanism,so as to maintain a good memory of the base class.Experimental results show that,in the three novel class splits of the PASCAL VOC,compared with the mainstream small sample object detection method FSODCR,the proposed algorithm achieves an average increased in nAP by 8.37 percentage points and 6.63 percentage points in the 2-Shot and 5-Shot settings respectively;on the MS COCO,compared with FSODCR,the proposed algorithm achieves an average increase in nAP and nAP75 of 1.2 percentage points and 0.9 percentage points respectively under the 10-Shot,and nAP and nAP75 have increased by 1.2 percentage points and 0.8 percentage points under the 30-Shot,which significantly improves the accuracy and stability of object detection under few-shot conditions.
李京;王子怡;巩海鑫;熊风光
中北大学 国际教育学院,山西 太原 030051中北大学 计算机科学与技术学院,山西 太原 030051中北大学 计算机科学与技术学院,山西 太原 030051中北大学 计算机科学与技术学院,山西 太原 030051||中北大学 机器视觉与虚拟现实山西省重点实验室,山西 太原 030051||中北大学 山西省视觉信息处理及智能机器人工程研究中心,山西 太原 030051
信息技术与安全科学
目标检测深度学习小样本目标检测迁移学习
object detectiondeep learningfew-shot object detectiontransfer learning
《中北大学学报(自然科学版)》 2026 (3)
263-273,11
国家自然科学基金(62272426)山西省科技重大专项计划"揭榜挂帅"项目(202201150401021)山西省自然科学基金(202203021212138,202303021211153,202203021222027)山西省科技成果转化引导专项(202104021301055)
评论