• Laser & Optoelectronics Progress
  • Vol. 61, Issue 8, 0812004 (2024)
Yuanpeng Wang1, Haibin Wan1、*, Kai Huang1, Zhaozhan Chi2, Jinqi Zhang1, and Zhixing Huang1
Author Affiliations
  • 1School of Computer, Electronics and Information, Guangxi University, Nanning 530004, Guangxi , China
  • 2School of Mechanical Engineering, Guangxi University, Nanning 530004, Guangxi , China
  • show less
    DOI: 10.3788/LOP231408 Cite this Article Set citation alerts
    Yuanpeng Wang, Haibin Wan, Kai Huang, Zhaozhan Chi, Jinqi Zhang, Zhixing Huang. Real-Time Detection of Abnormal Behavior of Escalator Passengers Based on YOLOv5s[J]. Laser & Optoelectronics Progress, 2024, 61(8): 0812004 Copy Citation Text show less

    Abstract

    To detect passengers' abnormal behavior in real time, we propose a lightweight escalator passenger' abnormal behavior real-time detection algorithm, YOLO-STE, based on YOLOv5s. First, a lightweight ShuffleNetV2 network was introduced in the backbone network to reduce the number of parameters and its computation. Second, a C3TR module based on Transformer encoding was introduced in the last layer of the backbone network to better extract rich global information and fuse features at different scales. Finally, an SE (Squeeze-and-excitation) attention mechanism was embedded in the feature fusion network of YOLOv5s to better focus on the main information and improve the model accuracy. We developed our dataset and conducted experiments. The experimental results demonstrate that compared with the original YOLOv5s, the mean Average Precision (mAP) of the improved algorithm is 1.9 percentage points higher, reaching 96.1%, and the model size is reduced by 70.8%. Moreover, the improved algorithm's forward propagation time is 39.9% shorter than that of the original YOLOv5s model when deployed and tested on the Jetson Nano hardware. Compared with the original YOLOv5s model, the improved algorithm can better achieve real-time detection of abnormal behavior of escalator passengers, which can better ensure the safety of passengers riding the escalator.
    Yuanpeng Wang, Haibin Wan, Kai Huang, Zhaozhan Chi, Jinqi Zhang, Zhixing Huang. Real-Time Detection of Abnormal Behavior of Escalator Passengers Based on YOLOv5s[J]. Laser & Optoelectronics Progress, 2024, 61(8): 0812004
    Download Citation