轻量级遥感目标检测模型,提升小物体识别精度与速度
RFWNet: A Lightweight Remote Sensing Object Detector Integrating Multiscale Receptive Fields and Foreground Focus Mechanism
- 融合多尺度感受野与前景聚焦机制,增强特征表达
- 在DOTA和NWPU数据集上分别达95.3%和73.2% mAP,仅600万参数
- 适合部署于资源受限的遥感实时检测场景
遥感目标检测面临类别相似度高、前景背景分布不均、目标尺寸小等挑战,且模型精度与计算复杂度间存在权衡。为此,本文提出一种高效轻量的遥感目标检测算法RFWNet,结合多尺度感受野与前景聚焦机制。设计轻量级主干网络RFASNet,利用遥感图像丰富的上下文信息提升类别可分性;提出前景-背景分离模块(FBSM),包含背景冗余信息过滤模块(BRIFM)与前景信息增强模块(FIEM),强化关键区域并抑制冗余背景;设计加权CIoU-Wasserstein损失(LWCW),通过归一化Wasserstein距离加权IoU损失,降低模型对小目标位置偏差的敏感性。实验表明,RFWNet在DOTA V1.0和NWPU VHR-10数据集上分别达到95.3%和73.2% mAP,参数量仅6.0 M,推理速度达52 FPS。
原文摘要 · Abstract (English)
Challenges in remote sensing object detection(RSOD), such as high interclass similarity, imbalanced foreground-background distribution, and the small size of objects in remote sensing images, significantly hinder detection accuracy. Moreover, the tradeoff between model accuracy and computational complexity poses additional constraints on the application of RSOD algorithms. To address these issues, this study proposes an efficient and lightweight RSOD algorithm integrating multiscale receptive fields and foreground focus mechanism, named robust foreground weighted network(RFWNet). Specifically, we proposed a lightweight backbone network receptive field adaptive selection network (RFASNet), leveraging the rich context information of remote sensing images to enhance class separability. Additionally, we developed a foreground-background separation module(FBSM)consisting of a background redundant information filtering module (BRIFM) and a foreground information enhancement module (FIEM) to emphasize critical regions within images while filtering redundant background information. Finally, we designed a loss function, the weighted CIoU-Wasserstein loss (LWCW),which weights the IoU-based loss by using the normalized Wasserstein distance to mitigate model sensitivity to small object position deviations. The comprehensive experimental results demonstrate that RFWNet achieved 95.3% and 73.2% mean average precision (mAP) with 6.0 M parameters on the DOTA V1.0 and NWPU VHR-10 datasets, respectively, with an inference speed of 52 FPS.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。