Bearing surface defect detection algorithm based on deep learning

Dehui Zhou; Jun Zhao; Jinfeng Cheng

doi:10.1117/12.3033566

13 June 2024 Bearing surface defect detection algorithm based on deep learning

Dehui Zhou, Jun Zhao, Jinfeng Cheng

Proceedings Volume 13180, International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2024); 131803Z (2024) https://doi.org/10.1117/12.3033566
Event: International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2024), 2024, Guangzhou, China

Abstract

Taking into account the industrial environment, where target detection algorithms aimed at tiny defects on the bearing surface suffer from low precision and slow speed, an improved RT-DETR algorithm, based on deep learning techniques is proposed. Firstly, the FasterNet is utilized to rebuild the backbone network, reducing computational stress while embedding a DWRF module integrated with EffectiveSE attention mechanism, allowing for in-depth feature extraction and bolstering finer-grained feature depiction, consequently reducing missed detections. Secondly, by incorporating Cascaded Group Attention (CGA) and designing an S-Fusion module embedded with SimAM attention, further limit computational redundancy and amplify the ability to capture key features. Finally, the NWD (normalized Wasserstein distance) combined with Inner-MPDIoU is applied for joint optimization loss, hastening the model's convergence and boosting the detection accuracy for small targets. The experimental results illustrate that in comparison to the original RT-DETR algorithm, the improved algorithm increases the mean average precision by 2.4%, reduces computation by 16.1%, and augments detection speed by 15.6%. Furthermore compared with Faster R-CNN, YOLOv5, YOLOv8 and other methods, our algorithm exhibits superior detection speed and accuracy, thereby meeting the requirements for detection of bearing surface tiny defects in industrial environment.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Dehui Zhou, Jun Zhao, and Jinfeng Cheng "Bearing surface defect detection algorithm based on deep learning", Proc. SPIE 13180, International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2024), 131803Z (13 June 2024); https://doi.org/10.1117/12.3033566

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
7 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Detection and tracking algorithms

Object detection

Defect detection

Feature extraction

Mathematical optimization

Deep learning

Feature fusion

Show All Keywords

Keywords/Phrases

Search In:

Publication Years