Slighter Faster R-CNN for real-time detection of steel strip surface defects

Effective surface defect detection methods are of great significance for the production of high quality steel strip. Aiming at real-time detection of steel strip surface defect, this paper constructed a slighter Faster R-CNN. Firstly, convolutional layers for feature extraction in Faster R-CNN were replaced by depthwise separable convolutions so that the speed of the network increased three to four times. Then, center loss was added to the original loss function to improve the network's ability to distinguish different types of defects. Finally, a surface defect dataset containing 4655 images of 6 classes was established, and the proposed networks were trained on it. The proposed networks achieved 98.32 % accuracy with an average speed of 0.05s per image. Experimental results show that the slighter Faster R-CNN outperforms other steel strip surface defect detection methods in both accuracy and speed.

[1]  Ali Farhadi,et al.  You Only Look Once: Unified, Real-Time Object Detection , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[2]  Dit-Yan Yeung,et al.  Learning a Deep Compact Image Representation for Visual Tracking , 2013, NIPS.

[3]  Yuanxiang Li,et al.  Classification of defects in steel strip surface based on multiclass support vector machine , 2014, Multimedia Tools and Applications.

[4]  Ali Farhadi,et al.  YOLO9000: Better, Faster, Stronger , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[5]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[6]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[7]  Maoxiang Chu,et al.  A Fast Detection Method for Region of Defect on Strip Steel Surface , 2015 .

[8]  Bo Chen,et al.  MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications , 2017, ArXiv.

[9]  Trevor Darrell,et al.  Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation , 2013, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[10]  Ross B. Girshick,et al.  Fast R-CNN , 2015, 1504.08083.

[11]  Yixin Yin,et al.  Real-time Detection of Steel Strip Surface Defects Based on Improved YOLO Detection Network , 2018 .

[12]  Jian Sun,et al.  Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition , 2014, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[13]  Yu Qiao,et al.  A Discriminative Feature Learning Approach for Deep Face Recognition , 2016, ECCV.

[14]  Liqiang Nie,et al.  Surface defect classification in large-scale strip steel image collection via hybrid chromosome genetic algorithm , 2016, Neurocomputing.

[15]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[16]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[17]  Kaiming He,et al.  Mask R-CNN , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[18]  Xu Ke,et al.  Surface Defect Recognition of Hot-rolled Steel Plates Based on Tetrolet Transform , 2016 .

[19]  Jie Zhao,et al.  Steel surface defects recognition based on multi-label classifier with hyper-sphere support vector machine , 2017, 2017 29th Chinese Control And Decision Conference (CCDC).

[20]  Stéphane Mallat,et al.  Rigid-Motion Scattering for Texture Classification , 2014, ArXiv.

[21]  F. Liou,et al.  Defect classification of laser metal deposition using logistic regression and artificial neural networks for pattern recognition , 2017, The International Journal of Advanced Manufacturing Technology.

[22]  Koen E. A. van de Sande,et al.  Selective Search for Object Recognition , 2013, International Journal of Computer Vision.

[23]  C. Lawrence Zitnick,et al.  Edge Boxes: Locating Object Proposals from Edges , 2014, ECCV.