Transactions of Nanjing University of Aeronautics & Astronautics
Attention Mechanism-Based Method for Intrusion Target Recognition in Railway
Author:
Affiliation:

1.CHN Energy ShuoHuang Railway Development Company Ltd, Beijing 100080, P. R. China;2.School of Mechanical and Electronic Control Engineering, Beijing Jiaotong University, Beijing 100044, P. R. China

Clc Number:

TN925

Fund Project:

This work was supported in part by the Science and Technology Innovation Project of CHN Energy ShuoHuang Railway Development Company Ltd (No.SHTL-22-28), the Beijing Natural Science Foundation-Fengtai Urban Rail Transit Frontier Research Joint Fund (No.L231002), and the Major Project of China State Railway Group Co., Ltd.(No.K2023T003).

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    The detection of foreign object intrusion is crucial for ensuring the safety of railway operations. To address challenges such as low efficiency, suboptimal detection accuracy, and slow detection speed inherent in conventional comprehensive video monitoring systems for railways, a railway foreign object intrusion recognition and detection system is conceived and implemented using edge computing and deep learning technologies. In a bid to raise detection accuracy, the convolutional block attention module (CBAM), including spatial and channel attention modules, is seamlessly integrated into the YOLOv5 model, giving rise to the CBAM-YOLOv5 model. Furthermore, the distance intersection-over-union_non-maximum suppression (DIoU_NMS) algorithm is employed in lieu of the weighted non-maximum suppression algorithm, resulting in improved detection performance for intrusive targets. To accelerate detection speed, the model undergoes pruning based on the batch normalization (BN) layer, and TensorRT inference acceleration techniques are employed, culminating in the successful deployment of the algorithm on edge devices. The CBAM-YOLOv5 model exhibits a notable 2.1% enhancement in detection accuracy when evaluated on a self-constructed railway dataset, achieving 95.0% for mean average precision (mAP). Furthermore, the inference speed on edge devices attains a commendable 15 frame/s.

    Reference
    Related
    Cited by
Get Citation

SHI Jiang, BAI Dingyuan, GUO Baoqing, WANG Yao, RUAN Tao. Attention Mechanism-Based Method for Intrusion Target Recognition in Railway[J]. Transactions of Nanjing University of Aeronautics & Astronautics,2024,(4):541-554

Copy
分享
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:May 07,2024
  • Revised:August 10,2024
  • Adopted:
  • Online: August 25,2024
  • Published:

WeChat

Mobile website