Laser & Optoelectronics Progress, Volume. 61, Issue 8, 0812004(2024)

Real-Time Detection of Abnormal Behavior of Escalator Passengers Based on YOLOv5s

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Instrumentation, Measurement and Metrology

    Received: May. 30, 2023

    Accepted: Jul. 24, 2023

    Published Online: Mar. 27, 2024

    The Author Email: Wan Haibin (hbwan@gxu.edu.cn)

    DOI:10.3788/LOP231408

    Topics