Qiu Hong, Yang Dayong, Cao Juanhua, Ming Jingqiang, Jiang Kun, Wu Weijun
School of Advanced Manufacturing, Nanchang University, Nanchang 330031, China.
Jiangxi Technical College of Manufacturing, Nanchang 330031, China.
Sensors (Basel). 2025 Sep 16;25(18):5753. doi: 10.3390/s25185753.
Aiming at the problems of excessive computational load, insufficient real-time performance, and an excessive amount of model parameters in track inspection, this paper proposes a lightweight track feature detection module (YOLO-LWTD) based on YOLO11n: first, the StarNet module is integrated into the backbone network, and its elemental multiplication operation is utilized to enhance the feature characterization capability; second, in the neck part, a lightweight extended path aggregation network reconstructs the feature pyramid information flow paths by combining with the C3K2-Light module to enhance the efficiency of the multi-scale feature fusion; finally, in the head part, a lighter and more efficient detection header, Detect-LADH, is used to reduce the feature decoding complexity. Experimental validation showed that the improved model outperforms the benchmark model in precision, recall, and mean average precision (MAP) by 0.5%, 2.0%, and 0.8%, respectively, with an inference speed of 163 FPS (a 38.1% improvement). The model volume is compressed to 1.5 MB (a 71.1% lightweight rate). This provides an energy-efficient solution for lightweight track detection tasks geared towards embedded deployment or real-time processing.
针对轨道检测中计算负载过大、实时性能不足以及模型参数量过多的问题,本文提出了一种基于YOLOv5的轻量化轨道特征检测模块(YOLO-LWTD):首先,将StarNet模块集成到主干网络中,利用其元素乘法运算增强特征表征能力;其次,在颈部,通过结合C3K2-Light模块,构建轻量化扩展路径聚合网络,重建特征金字塔信息流路径,提升多尺度特征融合效率;最后,在头部,采用更轻量高效的检测头Detect-LADH,降低特征解码复杂度。实验验证表明,改进后的模型在精度、召回率和平均精度均值(MAP)上分别比基准模型提高了0.5%、2.0%和0.8%,推理速度达到163 FPS(提高了38.1%)。模型体积压缩至1.5 MB(轻量化率达71.1%)。这为面向嵌入式部署或实时处理的轻量化轨道检测任务提供了节能解决方案。