首页|融合感受野扩增与特征增强的遥感小目标检测

融合感受野扩增与特征增强的遥感小目标检测

扫码查看
针对遥感图像小目标因背景复杂、尺寸较小以及排列密集等因素导致检测精度不高的问题,提出一种融合感受野扩增与特征增强的遥感小目标检测方法。该方法以YOLOv8s作为基线网络,首先对主干网络特征提取部分构造感受野扩增模块,通过双层路由注意力(BRA)高效捕获全局特征信息;其次在特征金字塔部分构建浅层特征融合结构,并在浅层特征图横向连接部分添加改进的坐标空间注意力(CSA),以增强小目标的特征信息;最后通过改进非极大值抑制(NMS)算法对检测结果进行后处理,以适应不同密度物体的检测。在DIOR遥感图像数据集上进行实验,预测框与真实框之间的交并比阈值(IoU)为 0。5 时的平均精度均值(mAP)达到90。3%,比原模型高出3%;IoU为 0。5∶0。95 时的mAP 达到 71。3%,比原模型高出 6。1%,实验结果表明,改进模型对遥感图像小目标检测任务具有较好的应用价值。
Remote sensing small target detection integrating receptive field amplification and feature enhancement
Aiming at the problem of low detection accuracy of small targets in remote sensing images due to com-plex background,small size and dense arrangement,a remote sensing small target detection method integrating recep-tive field amplification and feature enhancement is proposed.Using YOLOv8s as the baseline network,the method firstly constructs a receptive field amplification module for the feature extraction part of the backbone network,and effi-ciently captures the global feature information through the Bi-Level Routing Attention(BRA);secondly,it constructs a shallow feature fusion structure in the feature pyramid part,and adds the improved coordinate spatial attention(CSA)in the transverse connectivity part of the shallow feature map,in order to enhance the feature information of the small targets;Finally,the detection results are post-processed by an improved non-maximum suppression(NMS)al-gorithm to adapt to the detection of objects with different densities Experiments are carried out on the DIOR remote sensing image dataset,the mean average preci-sion(mAP)reaches 90.3%when the intersection and concurrency ratio threshold(IoU)between the predicted frame and the real frame is 0.5,which is 3%higher than that of the original model;and the mAP reaches 71.3%when the IoU is 0.5∶0.95,which is 6.1%higher than that of the original model,and the experimental re-sults show that the improved model has a good application value for the small target detection in remote sensing images.

receptive field amplificationfeature enhancementremote sensing small targetYOLOv8s

帖军、秦锦添、郑禄、郑明雪、陈婷

展开 >

中南民族大学计算机科学学院,武汉 430074

农业区块链与智能管理湖北省工程研究中心,武汉 430074

湖北省农业科学院农业经济技术研究所,武汉 430064

感受野扩增 特征增强 遥感小目标 YOLOv8s

2024

激光杂志
重庆市光学机械研究所

激光杂志

CSTPCD北大核心
影响因子:0.74
ISSN:0253-2743
年,卷(期):2024.45(12)