A Novel Gear Defect Detection Method Through Improved Yolov5 Network Using Attention Mechanism and Feature Fusion

Shuwen Zhang,Minqi He,Zhenyu Zhong,Dahu Zhu
DOI: https://doi.org/10.2139/ssrn.4161321
2022-01-01
SSRN Electronic Journal
Abstract:This work is motivated by the misdetected and undetected errors in metallic gear defect detection when the defect features are very close to the interferences of oil stains and image shadow. In this paper we present a novel gear surface defect detection method based on the improved YOLOv5 network model to address this problem. The presented method combines the approach using the attention module CBAMC3 constituted by CBAM and C3 modules together with the analysis of the module BiFPN_concat, which makes the network capable of extracting and fusing defect features, respectively. The cosine annealing function is then employed to improve the learning ability of the network by modifying the learning rate. The experimental results indicate that the improved YOLOv5 network enhances both the recall rate by 13.1% and the mAP@0.5 by 12% based on the constructed gear defect dataset compared with the original YOLOv5 network.
What problem does this paper attempt to address?