Bearing surface defect detection based on improved convolutional neural network
Authors:
Open Access
- 1 January 2023
- journal article
- research article
- Published by American Institute of Mathematical Sciences (AIMS) in Mathematical Biosciences and Engineering
- Vol. 20 (7) , 12341-12359
Abstract
This paper addresses the issue of artificial visual inspection being overly reliant on subjective experience and the difficulty for the human eye to accurately identify dense and non-significant defects. To solve this problem, we have implemented an automatic object detection algorithm based on an improved version of YOLOv5.First, we use the K-means++ clustering algorithm to automatically calculate the Anchor of the model to reduce the effect of the close location of the initial clustering centers on the clustering of the sample data.Second, we add the Coordinate Attention (CA) attention mechanism to the model to allow the model to better capture and understand important features in the images. Then, we add a new detection layer with a downsampling multiplier of 4 to the Neck network to improve the precision of the model. Finally, we use the lightweight network MobileNetV3 instead of YOLOv5's backbone network to reduce the model detection time overhead.Our model achieves 85.87% mAP, which is 6.44% better than the YOLOv5 network, and the detection time for a single image is only 54ms, which is 50% faster than the YOLOv5 network. After testing, we have proven that our proposed algorithm can quickly and accurately detect the condition of bearing appearance defects, improving detection efficiency and reducing costs.Keywords
This publication has 11 references indexed in Scilit:
- Application of Convolutional Neural Network for Fault Diagnosis of Bearing Scratch of an Induction MotorApplied Sciences, 2022
- Feature Enhancement Based on Regular Sparse Model for Planetary Gearbox Fault DiagnosisIEEE Transactions on Instrumentation and Measurement, 2022
- Effect of data preprocessing methods and hyperparameters on accuracy of ball bearing fault detection based on deep learningAdvances in Mechanical Engineering, 2022
- Improved faster R-CNN for fabric defect detection based on Gabor filter with Genetic Algorithm optimizationComputers in Industry, 2022
- FPCB Surface Defect Detection: A Decoupled Two-Stage Object Detection FrameworkIEEE Transactions on Instrumentation and Measurement, 2021
- Physics-Based Convolutional Neural Network for Fault Diagnosis of Rolling Element BearingsIEEE Sensors Journal, 2019
- Defect Diagnosis of Rolling Element Bearing using Deep Learning2018
- Multiple Fault Diagnosis of Electric Powertrains Under Variable Speeds Using Convolutional Neural Networks2018
- Simulation-driven machine learning: Bearing fault classificationMechanical Systems and Signal Processing, 2018
- Fault diagnosis of ball bearings using machine learning methodsExpert Systems with Applications, 2011