胡波,刘培文,刘瑞琪
(广西科技大学自动化学院,广西 柳州 545006)
摘要:针对城市列车轨道区域与路面背景相似度较高而导致的轨道区域提取效果较差的问题,提出一种基于改进DeeplabV3+模型的轨道区域的分割算法.首先,针对列车轨道区域长宽比例较大的特点,使用条形池化卷积层代替ASPP结构中的全局池化层,以提升算法对条形轨道区域的提取能力;其次设计使用特征分组注意力(FGA)机制,提升模型对细节的关注度;最后改变上采样频率,降低因快速上采样而造成边界信息的模糊化影响.将3 192张城市轨道图像作为数据集,经验证与原网络相比,改进网络在数据集中的平均交并比以及召回率分别提高了3.65%、3.83%,达到93.88%以及97.35%,有效提升了轨道区域提取效果.
关键词:区域提取;Deeplabv3+;条形池化;注意力;空间金字塔
中图分类号:TP391.4;U212.24
文献标识码:A
文章编号:1673-2103(2025)05-0076-07
收稿日期:2025-01-03
作者简介:胡波(1981— ),男,广西柳州人,副教授,硕士,研究方向:机器视觉、数据挖掘、智能控制,E-mail:443722152@qq.com.
通信作者:刘瑞琪(1980— ),男,甘肃兰州人,高级工程师,研究方向:故障诊断、深度学习、自然语言处理、机器视觉,E-mail:1925930700@qq.com.