Abstract:The automatic detection of foreign objects between platform screen doors (PSDs) and metro train doors significantly affects personnel and property safety and maintains the train's normal operation. However, some existing works only determine the presence of foreign objects but cannot indicate their categories. Besides, although deep-learning-based object detection algorithms can indicate the presence and categories of foreign objects, most of them only harness the information in region proposals, ignoring global contextual information. Furthermore, their performance comes at the considerable cost of computational complexity, and leading cannot be well deployed in the metro environment. To address these issues and better implement foreign object detection (FOD), we present You Only Look Once-Transformer (YOLO-Former), a simple but efficient model. YOLO-Former is accomplished based on YOLOv5 through the following procedure. First, the vision transformer (ViT) is introduced for dynamic attention and global modeling, thereby solving the problem that the original YOLOv5 only utilizes information in region proposals and has insufficient ability to capture global information. Second, the convolutional block attention module (CBAM) and the stem module are used to improve feature expression ability further and reduce floating-point operations (FLOPs). Finally, we design various variants with different widths and depths to meet every need. Experiments on the FOD dataset (FODD) and the PASCAL VOC dataset demonstrate that YOLO-Former-x consistently outperforms other state of the arts with significant margins (0.5-11.3 mean average precision (mAP) on FODD and 0.6-13.6 on the PASCAL VOC dataset). Last but not least, YOLO-Former-x maintains real-time processing speed (27.32 and 28.17 frames/s (FPS) on TITAN Xp).

An Improved YOLOv5 for Transformer Nameplate Text Detection

An Object Detection Method Based on Improved YOLOX

Improved YOLOV5 Angle Embossed Character Recognition by Multiscale Residual Attention with Selectable Clustering

Efficient Vision Transformer YOLOv5 for Accurate and Fast Traffic Sign Detection

An improved YOLOv5 algorithm for obscured target recognition

STOE-YOLOv5s: A Hard Hat Wearing Detection Algorithm Based on Improved YOLOv5s

YOLO-Extract: Improved YOLOv5 for Aircraft Object Detection in Remote Sensing Images

4AC-YOLOv5: an improved algorithm for small target face detection

Improvement and Enhancement of YOLOv5 Small Target Recognition Based on Multi-module Optimization

YOLOv5 Algorithm Optimization: Precise Target Detection Amidst Background Interference

An Improved YOLOv5s Algorithm for Object Detection with an Attention Mechanism

An improved YOLOv5 method for large objects detection with multi-scale feature cross-layer fusion network

TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios

Small Target Detection Optimization Algorithm Based on Improved YOLOv5

Improved YOLOv5 with Transformer for Large Scene Military Vehicle Detection on SAR Image

YOLO-SDH: improved YOLOv5 using scaled decoupled head for object detection

HIC-YOLOv5: Improved YOLOv5 For Small Object Detection

CNTR-YOLO: Improved YOLOv5 Based on ConvNext and Transformer for Aircraft Detection in Remote Sensing Images

YOLO-Former: Marrying YOLO and Transformer for Foreign Object Detection

YOLO-DD: Improved YOLOv5 for Defect Detection

YOLOFM: an improved fire and smoke object detection algorithm based on YOLOv5n