基于改进DeepLabV3+的河流岸线提取轻量化算法

    Lightweight algorithm for river shoreline extraction based on improved DeepLabV3+

    • 摘要: 监测河流区域和河流岸线对于水文站至关重要,能够为水资源管理、防洪预警和生态保护提供重要数据支持。传统监测方法存在效率低下、精度不足等问题。因此,提出了一种基于MobileNetV2骨干网络的轻量级DeepLabV3+网络。该网络通过减少模型参数数量,显著提高了运算速度;同时引入Dice Loss作为损失函数,优化了模型在河岸线边缘细节和小规模区域分割任务中的表现;此外,模型结合了空间注意力模块(Spatial Attention Module,SAM)和带有Squeeze-and-Excitation(SE)注意力机制的空洞空间金字塔池化(Atrous Spatial Pyramid Pooling with Squeeze-and-Excitation,ASPP-SE),进一步增强了对复杂环境中河流特征的捕捉能力。实验结果表明:改进模型的Kappa系数、Dice系数、F1-Score系数分别提升了10.43%,3.41%,0.02%,优于原DeepLabV3+网络以及U-Net、PSPNet和HRNet,即改进网络在复杂条件下提取河流区域和河流岸线时具有更高的准确性和效率。研究成果可为水文站的河流监测工作提供一种高效、准确的技术手段支持。

       

      Abstract: Monitoring river areas and shorelines is crucial for hydrological stations, as it provides essential data support for water resource management, flood prevention warnings, and ecological conservation.Traditional monitoring methods suffer from inefficiency and insufficient accuracy.Therefore, a lightweight DeepLabV3+ network based on the MobileNetV2 backbone is proposed.By reducing the number of model parameters, this network significantly improves computational speed.Additionally, the introduction of Dice Loss as the loss function optimizes the model′s performance in segmenting river shoreline edge details and small-scale regions.Furthermore, this model incorporates a Spatial Attention Module (SAM) and an Atrous Spatial Pyramid Pooling with Squeeze-and-Excitation (ASPP-SE), further enhancing its ability to capture river features in complex environments.Experimental results show that the improved model achieves increases of 10.43%, 3.41%, and 0.02% in the Kappa coefficient, Dice coefficient, and F1-Score, respectively, outperforming the original DeepLabV3+ network as well as U-Net, PSPNet, and HRNet.This indicates that the improved network offers higher accuracy and efficiency in extracting river areas and shorelines under complex conditions.The research findings provide an efficient and accurate technical solution to support river monitoring tasks at hydrological stations.

       

    /

    返回文章
    返回