arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2503.00545cs.CV

RFWNet:一种融合多尺度感受野与前景聚焦机制的轻量级遥感目标检测器

RFWNet: A Lightweight Remote Sensing Object Detector Integrating Multiscale Receptive Fields and Foreground Focus Mechanism

China Agricultural University(中国农业大学);Shenzhen University(深圳大学)

Yujie Lei, Wenjie Sun, Sen Jia, Qingquan Li, Jie Zhang

AI总结:

针对遥感目标检测中类间相似度高、前景背景不平衡和小目标问题,本文提出轻量级RFWNet,通过多尺度感受野骨干网络、前景-背景分离模块及加权CIoU-Wasserstein损失,在保持低参数量的同时实现高精度检测。

摘要:

遥感目标检测(RSOD)面临诸多挑战,例如类间相似度高、前景与背景分布不平衡以及遥感图像中目标尺寸小,这些因素显著制约了检测精度。此外,模型精度与计算复杂度之间的权衡对RSOD算法的应用构成了额外限制。为解决上述问题,本研究提出了一种融合多尺度感受野与前景聚焦机制的高效轻量级RSOD算法,命名为鲁棒前景加权网络(RFWNet)。具体而言,我们提出了轻量级骨干网络——感受野自适应选择网络(RFASNet),利用遥感图像丰富的上下文信息增强类别可分性。此外,我们开发了前景-背景分离模块(FBSM),该模块由背景冗余信息过滤模块(BRIFM)和前景信息增强模块(FIEM)组成,用于突出图像中的关键区域,同时过滤冗余背景信息。最后,我们设计了一种损失函数,即加权CIoU-Wasserstein损失(LWCW),该损失通过使用归一化Wasserstein距离对基于IoU的损失进行加权,以降低模型对小目标位置偏差的敏感性。综合实验结果表明,RFWNet在DOTA V1.0和NWPU VHR-10数据集上分别以6.0 M参数量取得了95.3%和73.2%的平均精度(mAP),推理速度为52 FPS。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑