计算机工程

作者：李仲, 冒睿瑞, 王晓龙, 王根一, 安国成

单位：1. 上海华讯网络系统有限公司服务运作部, 上海 2011032. 中国电子科技集团公司第三十二研究所, 上海 201808

关键词：语义分割,水位线检测,金字塔池化模块,注意力,多场景

基金：国家重点研发计划(2023YFC3006700)

PIDNet是三分支网络构成的语义分割模型, 在众多竞赛数据集中均保持优秀的分割精度。然而, 积分分支中进行多次下采样和金字塔池化模块中多分支特征融合冗余的缺点限制了算法精度的提高。在水位线检测任务中, 现有算法的缺点会导致局部细节信息丢失, 使得水体边缘精细化检测的能力有所下降。为了缓解这个问题, 提出一种基于改进PIDNet的水位线检测算法。首先设计一种结合通道注意力的轻量化像素增强模块, 在积分分支下采样过程中进行像素增强, 减少局部信息丢失。然后对金字塔池化模块进行重构, 在减小池化输出特征大小的基础上减少并行分支数, 同时在特征融合时结合通道注意力进一步加强关注重要特征的能力, 提高水位线边缘的分割精度。此外, 融合多场景的河流数据集, 避免复杂场景下检测出的水位线位置发生偏移和断线。实验结果表明, 所提方法(S和M)在水位线检测任务中相对原算法(S和M)在3个性能指标上都有所提高, 以M规模为例, 像素正确率提高了1.47个百分点, 平均交并比提高了1.04个百分点, 检测延迟降低了0.9 ms。

来源：2024年第8期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第8期