Improved Real-Time Detection Transformer-Based Rail Fastener Defect Detection Algorithm

Wei Song, Bin Liao, Keqing Ning () and Xiaoyu Yan
Additional contact information
Wei Song: School of Information, North China University of Technology, Beijing 100144, China
Bin Liao: School of Information, North China University of Technology, Beijing 100144, China
Keqing Ning: School of Information, North China University of Technology, Beijing 100144, China
Xiaoyu Yan: School of Information, North China University of Technology, Beijing 100144, China

Mathematics, 2024, vol. 12, issue 21, 1-14

Abstract: To address the issues of the Real-Time DEtection TRansformer (RT-DETR) object detection model, including poor defect feature extraction in the task of rail fastener defect detection, inefficient use of computational resources, and suboptimal channel attention in the self-attention mechanism, the following improvements were made. Firstly, a Super-Resolution Convolutional Module (SRConv) was designed as a separate component and integrated into the Backbone network, which enhances the image details and clarity while preserving the original image structure and semantic content. This integration improves the model’s ability to extract defect features. Secondly, a channel attention mechanism was integrated into the self-attention module of RT-DETR to enhance the focus on feature map channels, addressing the problem of sparse attention maps caused by the lack of channel attention while saving computational resources. Finally, the experimental results show that compared to the original model, the improved RT-DETR-based rail fastener defect detection algorithm, with an additional 0.4 MB of parameters, achieved a higher accuracy, with a 2.8 percentage point increase in the Mean Average Precision (mAP) across IoU thresholds from 0.5 to 0.9 and a 1.7 percentage point increase in the Average Recall (AR) across the same thresholds.

Keywords: transformer; super-resolution; fastener; track; defect detection (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/21/3349/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/21/3349/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:12:y:2024:i:21:p:3349-:d:1506748

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().