煤炭定量装车站智能化装车需要机车慢速且恒速运行，目前煤炭装车系统完全依靠人工通过对讲机以喊话的方式来告知机车驾驶员控制车速。随着人工智能技术的发展，在装煤机车控制中，可以实现自动加速。其中，自动驾驶首要任务就是判断前方可行驶区域。受UNet和Segnet网络启发，设计了一种基于AI语义分割技术的铁轨分割模型。首先采集列车视角的铁轨数据，对数据预处理和标注，制作装车站环境下的铁轨数据集。然后构建Encode-Decode结构的神经网络，Encode阶段以卷积残差网络为主干网络，提取并融合不同层级的特征图。Decode阶段结合特征图和最大池化索引通过转置卷积恢复图像细节特征。最后通过两阶段随机梯度下降优化算法训练网络，得到铁轨分割模型。网络模型训练完毕以后，在数据集上进行对比试验，实验结果表明，该方法相对传统图像检测的方法可以有效分割出铁轨区域，模型泛化能力较强。

1 背 景
2 ResUNet网络和两阶段优化方法
2.1 铁轨分割网络模型
2.2 基于最大池化索引的上采样
2.3 两阶段随机梯度下降算法
3 实 验
4 结 语

豆鹏.基于AI语义分割的内燃机车自动驾驶在煤炭装车中的研究[J].煤炭加工与综合利用,2022(12):60-64.DOI:10.16200/j.cnki.11-2627/td.2022.12.014.