Deep learning-based fast recognition of commutator surface defects

被引:28
作者
Shu, Yu Feng [1 ,2 ]
Li, Bin [2 ]
Li, Xiaomian [1 ]
Xiong, Changwei [1 ]
Cao, Shenyi [1 ]
Wen, Xin Yan [1 ]
机构
[1] DongGuan Polytech, Dept Mech & Elect Engn, Dongguan 523808, Peoples R China
[2] HUST, Sch Mech Sci & Engn, State Key Lab Digital Mfg Equipment & Technol, Wuhan 430074, Peoples R China
关键词
Surface defects; Target detection algorithm; Convolutional neural network; Deep learning;
D O I
10.1016/j.measurement.2021.109324
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
With low accuracy and poor efficiency, the manual and traditional detection of commutator surface defects cannot meet efficiency and timeliness requirements. Thus, a method is proposed to detect and identify commutator surface defects. YOLOv3 target detection is applied to commutator surface defect detection and recognition, and the model size and parameter number are reduced without significantly reducing detection accuracy. First, this paper proposes a separable residual module based on deep separable convolutions and residual networks. A network with shallower layers and fewer channels is designed for quick detection and recognition of commutator surface defects. Finally, the algorithm herein is evaluated on SD_data. The experimental results indicate that while maintaining good accuracy, this method has a smaller model size, fewer parameters and faster running time than the YOLOv3 network. Moreover, accuracy is improved to a certain extent compared to the current commutator surface defect detection methods, so it can be better applied to real-time detection and recognition of commutator surface defects.
引用
收藏
页数:8
相关论文
共 26 条
[1]  
Adam A.G., 2017, ARXIV170404861
[2]  
[Anonymous], 2008, PROC IEEE C COMPUTER
[3]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893
[4]   The PASCAL Visual Object Classes Challenge: A Retrospective [J].
Everingham, Mark ;
Eslami, S. M. Ali ;
Van Gool, Luc ;
Williams, Christopher K. I. ;
Winn, John ;
Zisserman, Andrew .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 111 (01) :98-136
[5]  
Freund Y., 1996, Machine Learning. Proceedings of the Thirteenth International Conference (ICML '96), P148
[6]  
Girshick R, 2015, P ADV NEUR INF PROC, P127
[7]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[8]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[9]  
Government information disclosure column, 2015, NOT STAT COUNC PRINT
[10]  
Hao R., 2017, CONTROL DECIS, V32, P1345