The task of instance segmentation in remote sensing images, aiming at performing per-pixel labeling of objects at instance level, is of great importance for various civil applications. Despite previous successes, most existing instance segmentation methods designed for natural images encounter sharp performance degradations when they are directly applied to top-view remote sensing images. Through careful analysis, we observe that the challenges mainly come from the lack of discriminative object features due to severe scale variations, low contrasts, and clustered distributions. In order to address these problems, a novel context aggregation network (CATNet) is proposed to improve the feature extraction process. The proposed model exploits three lightweight plug-and-play modules, namely dense feature pyramid network (DenseFPN), spatial context pyramid (SCP), and hierarchical region of interest extractor (HRoIE), to aggregate global visual context at feature, spatial, and instance domains, respectively. DenseFPN is a multi-scale feature propagation module that establishes more flexible information flows by adopting inter-level residual connections, cross-level dense connections, and feature re-weighting strategy. Leveraging the attention mechanism, SCP further augments the features by aggregating global spatial context into local regions. For each instance, HRoIE adaptively generates RoI features for different downstream tasks. Extensive evaluations of the proposed scheme on iSAID, DIOR, NWPU VHR-10, and HRSID datasets demonstrate that the proposed approach outperforms state-of-the-arts under similar computational costs. Source code and pre-trained models are available at https://github.com/yeliudev/CATNet.
翻译:遥感图像中的实例分割任务旨在对物体进行实例级别的逐像素标记,在各类民用应用中具有重要意义。尽管已有成功案例,但大多数为自然图像设计的现有实例分割方法在直接应用于俯视遥感图像时,性能出现显著下降。通过深入分析,我们观察到主要挑战源于尺度剧烈变化、对比度低以及分布密集导致的判别性物体特征不足。为解决这些问题,本文提出一种新型上下文聚合网络(CATNet)来改进特征提取过程。该模型利用三个轻量级即插即用模块——密集特征金字塔网络(DenseFPN)、空间上下文金字塔(SCP)和层次化兴趣区域提取器(HRoIE),分别在特征域、空间域和实例域聚合全局视觉上下文。DenseFPN是一种多尺度特征传播模块,通过引入层间残差连接、跨层密集连接和特征重加权策略,建立更灵活的信息流。借助注意力机制,SCP通过将全局空间上下文聚合到局部区域来进一步增强特征。对于每个实例,HRoIE自适应地为不同下游任务生成兴趣区域特征。在iSAID、DIOR、NWPU VHR-10和HRSID数据集上的广泛评估表明,所提方法在相似计算成本下优于现有最先进技术。源代码与预训练模型已公开于https://github.com/yeliudev/CATNet。