RSDNet: A New Multiscale Rail Surface Defect Detection Model

Bibliographic Details
Title: RSDNet: A New Multiscale Rail Surface Defect Detection Model
Authors: Jingyi Du, Ruibo Zhang, Rui Gao, Lei Nan, Yifan Bao
Source: Sensors, Vol 24, Iss 11, p 3579 (2024)
Publisher Information: MDPI AG, 2024.
Publication Year: 2024
Collection: LCC:Chemical technology
Subject Terms: rail surface defect detection, YOLOv8, CDConv, BiFPN, EMA, Chemical technology, TP1-1185
More Details: The rapid and accurate identification of rail surface defects is critical to the maintenance and operational safety of the rail. For the problems of large-scale differences in rail surface defects and many small-scale defects, this paper proposes a rail surface defect detection algorithm, RSDNet (Rail Surface Defect Detection Net), with YOLOv8n as the baseline model. Firstly, the CDConv (Cascade Dilated Convolution) module is designed to realize multi-scale convolution by cascading the cavity convolution with different cavity rates. The CDConv is embedded into the backbone network to gather earlier defect local characteristics and contextual data. Secondly, the feature fusion method of Head is optimized based on BiFPN (Bi-directional Feature Pyramids Network) to fuse more layers of feature information and improve the utilization of original information. Finally, the EMA (Efficient Multi-Scale Attention) attention module is introduced to enhance the network’s attention to defect information. The experiments are conducted on the RSDDs dataset, and the experimental results show that the RSDNet algorithm achieves a mAP of 95.4% for rail surface defect detection, which is 4.6% higher than the original YOLOv8n. This study provides an effective technical means for rail surface defect detection that has certain engineering applications.
Document Type: article
File Description: electronic resource
Language: English
ISSN: 1424-8220
Relation: https://www.mdpi.com/1424-8220/24/11/3579; https://doaj.org/toc/1424-8220
DOI: 10.3390/s24113579
Access URL: https://doaj.org/article/6274b05994ec43b48984dddd8eb7252c
Accession Number: edsdoj.6274b05994ec43b48984dddd8eb7252c
Database: Directory of Open Access Journals
Full text is not displayed to guests.
More Details
ISSN:14248220
DOI:10.3390/s24113579
Published in:Sensors
Language:English