基于改进YOLOv8的轨道小尺度异物入侵算法研究

打开文本图片集

摘  要： 针对当前列车轨道障碍物检测方法存在的小目标检测精度低、模型过大且部署成本高等问题，文中提出一种改进的YOLOv8⁃SGFE轨道侵限物检测模型。首先，为了减少网络的计算量，在小目标检测模块SPD⁃Conv的基础上，设计了一个SGConv模块，并用其替换YOLOv8主干层中的普通卷积层；其次，为了增强模型的感知能力，将高效多尺度注意力EMA与C2f⁃Faster模块相结合，构成C2f⁃Faster⁃EMA模块，并用其替换YOLOv8中的C2f模块；最后，将改进后的YOLOv8⁃SGFE模型应用于自制的铁路轨道侵限物数据集。（剩余10326字）