3 个月前

DecoupleNet:一种用于遥感视觉任务的轻量级主干网络及高效特征解耦机制

DecoupleNet:一种用于遥感视觉任务的轻量级主干网络及高效特征解耦机制

摘要

在计算机视觉(CV)领域,如何在速度与精度之间取得平衡仍是亟待解决的关键挑战。近年来,研究重点集中于开发轻量级网络,以优化计算效率并提升特征提取能力。然而,在遥感(RS)图像分析中,由于小目标与多尺度目标检测至关重要,现有轻量级网络在性能上往往表现不足。为此,本文提出一种名为DecoupleNet的创新轻量级主干网络,专为资源受限环境下的遥感视觉任务设计。DecoupleNet引入两个核心模块:特征融合下采样(FID)模块与多分支特征解耦(MBFD)模块。其中,FID模块在下采样过程中有效保留小目标的特征信息,而MBFD模块则通过一种新颖的解耦机制,显著增强小目标及多尺度目标的特征表达能力。在三个典型的遥感视觉任务上的全面评估表明,DecoupleNet在精度与计算效率之间展现出优于现有轻量级网络的卓越平衡性。在NWPU-RESISC45图像分类数据集上,DecoupleNet达到95.30%的Top-1准确率,较FasterNet提升2%,同时参数量更少、计算开销更低。在DOTA 1.0测试集上的目标检测任务中,DecoupleNet的检测准确率达到78.04%,优于ARC-R50的0.69%。在LoveDA数据集的语义分割任务中,DecoupleNet实现53.1%的准确率,较UnetFormer提升0.70%。上述结果为在资源受限设备上推进遥感图像分析提供了新的技术路径,填补了该领域的重要空白。相关代码与预训练模型已公开发布于GitHub:https://github.com/lwCVer/DecoupleNet。

基准测试

基准方法指标
image-classification-on-resisc45DecoupleNet D2
Top 1 Accuracy: 95.87
object-detection-in-aerial-images-on-dior-rDecoupleNet D2
mAP: 67.08
object-detection-in-aerial-images-on-dota-1DecoupleNet D2
mAP: 78.04%
semantic-segmentation-on-lovedaDecoupleNet D2
Category mIoU: 53.1
semantic-segmentation-on-uavidDecoupleNet D2
Mean IoU: 65.8

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DecoupleNet:一种用于遥感视觉任务的轻量级主干网络及高效特征解耦机制 | 论文 | HyperAI超神经