基于上下文与特征交互的小目标检测算法

无人机航拍图像内,小目标像素少、细节模糊且特征易丢失,现有算法对小目标检测效果不够理想。为此,提出一种基于上下文与特征交互的小目标检测算法。首先,采用ConvNeXt新一代卷积网络以增强特征表达能力;其次,设计稠密式信息增强模块(Dense Information Enhancement Module,DIEM)提取上下文信息;最后,提出特征交互单元(Feature Interaction Unit,FIU)促进浅层和深层特征的信息交互。检测结果显示,模型精度mAP50达到45.9%,相较于基线提升了6.0%,减少了错检漏检现象。