孟凡腾1,秦勇1,崔京1,吴云鹏2,张紫城1,魏少伟3
摘要: 铁路外部环境的常见隐患以及未知风险(包括泥石流、落石、动物入侵等)严重威胁到铁路安全运行,需要巡检人员耗时费力地频繁检查,但巡检范围仍非常有限。目前,低空经济已成为国家新质生产力代表,无人机具备高空、远距离、受地形和铁路维修天窗影响小,具有先天巡检优势。针对铁路外部环境未知风险样本稀疏且具有随机不确定性的挑战,本文利用无人机进行沿线遥感图像采集,并基于Faster R-CNN提出了一种未知风险检测框架。首先,设计了一种新颖的目标性与多分类解耦训练策略,并融合在未知风险检测框架中,显著提升了通用目标检测性能,避免将未知风险目标错分为背景;其次,改进了VOS的虚拟特征合成方法,提出了基于相似度的特征空间采样,在构建实例级目标特征空间基础上进行多元高斯分布参数估计与重采样,获得泛化性的未知风险目标特征表示;随后,利用基于能量的不确定性度量,对实例级特征进行不确定性度量,并据此计算损失以诱导网络优化常见类别和未知风险类别的决策边界。最终,在采集的铁路外部环境数据集、开源无人机数据集以及泛化性测试数据上进行了定量与定性实验分析,本文提出的方法在常见隐患识别上取得了95.7% mAP50,同时在测试集和泛化性测试数据上分别取得了98%和80.8% Recall50,实验结果表明本文方法在保证常见隐患类别高识别率的基础上对于未知风险目标也有较高的检测能力。
中图分类号: