基于改进YOLOv8s的铁路车站信号平面布置图信息提取方法
中国铁道科学
页数: 12 2024-09-15
摘要: 针对车站信号平面布置图在2次利用过程中,无法将不同设备图元转化为通用数据的问题,提出基于改进YOLOv8s的信息提取方法。首先,将空间和通道重建卷积(SCConv)与颈部网络中的C2f单元相结合,以降低特征图的冗余度,减少空间和通道维度上的重复信息,实现检测模型的轻量化;其次,在传统YOLOv8s模型的主干和颈部网络中引入高效多尺度注意力模块(EMA),以增强模型对图内设备周边... (共12页)