Laser & Optoelectronics Progress, Volume. 60, Issue 22, 2215005(2023)
Gear Surface Defect Detection Method Based on Improved YOLOx Network
Herein, an improved YOLOx algorithm is proposed to address the challenges concerning false and missing detection of metal gear surface defects in an industrial interference environment. First, by utilizing the adaptive spatial feature fusion (ASFF) to fully utilize the differences between the features of defects and interference items at different scales, the model’s anti-interference ability is improved. Second, through the effective channel attention (ECA) module, the network’s feature extraction capability is increased. Finally, the confidence loss function is modified to the Varifocal loss function, which reduces the interference of complex samples in the network. Experimental results indicate that the improved YOLOx network outperforms the original network. Particularly, the recall rate, accuracy, and mean average precision indexes of the improved YOLOx network are improved by 6.1, 4.6, and 9.4 percentage points, respectively, as compared with the original network.
Get Citation
Copy Citation Text
Shuwen Zhang, Zhenyu Zhong, Dahu Zhu. Gear Surface Defect Detection Method Based on Improved YOLOx Network[J]. Laser & Optoelectronics Progress, 2023, 60(22): 2215005
Category: Machine Vision
Received: Jan. 9, 2023
Accepted: Mar. 6, 2023
Published Online: Nov. 16, 2023
The Author Email: Zhu Dahu (dhzhu@whut.edu.cn)