For the current problems of low accuracy and poor reliability of defect detection of bearing roller end surfaces in industrial production, this paper proposes a bearing roller end surface defect detection algorithm based on improved YOLOv5n and the fusion of gamma-corrected maps and curvature maps. First, this paper uses photometric stereo to reconstruct the three-dimensional shape of the surface and proposes an improved Frankot-Chellappa integration algorithm to solve the problem of reconstructing surface deformation. Secondly, the DenseFuse network is used to fuse gamma-corrected maps and curvature maps to generate an image dataset that combines the strengths of both images to enhance defect features and improve the precision of target detection. Finally, the improved target detection model YOLOv5n is proposed to detect defects in the end surfaces of bearing rollers. The experimental results show that using fused images for training, detection models with higher mAP than traditional images can be obtained, and the improved YOLOv5n algorithm maintains the high real-time performance of the original algorithm while the mean average precision mAP0.5 and mAP0.5:0.95 of improved YOLOv5n are 98.6% and 87.4% respectively, which are 0.9% and 2.8% higher than YOLOv5n respectively.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.
customersupport@researchsolutions.com
10624 S. Eastern Ave., Ste. A-614
Henderson, NV 89052, USA
This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
Copyright © 2024 scite LLC. All rights reserved.
Made with 💙 for researchers
Part of the Research Solutions Family.