基于可分离残差与边缘加强的轻量级遥感图像语义分割

    Lightweight Semantic Segmentation of Remote Sensing Image Based on Separable Residuals and Edge Enhancement

    • 基于深度学习的方法已在高分辨率遥感图像的语义分割中取得较高精度,但存在参数量大、计算成本高的问题。为此,文章提出了一种轻量级边缘信息强化的高分辨率遥感图像语义分割网络。该网络采用轻量级网络MobileNetv3作为主干特征提取器,通过设计残差双通道特征提取模块改进瓶颈块,同时还引入了上下文特征增强器和深度可分离卷积,有效增强了特征信息的融合能力;针对遥感分割中的多尺度问题,设计了一种池化融合模块,实现了深层特征与浅层特征之间的信息互补。为验证该方法的有效性并评估其性能优势,文章将其与U-Net、DeeplabV3+等6种分割模型在ISPRS Potsdam和Vaihingen两个公共数据集上进行了对比实验。实验结果表明,该模型的的平均交并比在两个数据集上分别达到78.16%和76.78%,推理时间仅为0.75 s,各项指标均优于对比模型,在分割精度和计算效率之间取得了良好的平衡。

       

      Abstract: The semantic segmentation of high-resolution remote sensing images has achieved notable precision through deep learning methods, albeit demanding a substantial number of parameters and computations. To address the challenge of reducing computational costs without sacrificing precision, this paper advocates a lightweight edge-enhanced network for semantic segmentation of high-resolution remote sensing images. Initially, the network employs the lightweight MobileNetv3 as its backbone feature extractor. The bottleneck block is enhanced by designing a residual dual-channel feature extraction module. Additionally, a contextual feature enhancer and depthwise separable convolution are introduced, effectively strengthening the feature fusion capability. To address the multi-scale challenge in remote sensing segmentation, a pooling fusion module is proposed, achieving complementary information exchange between deep and shallow features.To validate the effectiveness of the proposed method and demonstrate its superiority, we evaluate it against six segmentation models, including U-Net and DeepLabV3+, on public dataset ISPRS Potsdam and Vaihingen. Experimental results demonstrate that the proposed model achieves mean Intersection over Union (mIoU) scores of 78.16% and 76.78% on the two datasets, with an inference time of only 0.75 seconds. All evaluation metrics outperform the competing models, indicating a favorable balance between segmentation accuracy and computational efficiency.

       

    /

    返回文章
    返回