Joints Steering Gear Detection Based on Semantic Segmentation

Aiming at the problem of inaccurate detection of joints steering gears in the industrialized inspection of automobiles, a method for joints steering gear detection based on image semantic segmentation is proposed. In this method, considering that this learning tasks are usually easy to overfit, appropriate regularization is needed to generalize well. In data augmentation, we randomly cover the input square area and randomly fill the corresponding square areas of other data into the area which is covered. Adopting encoder-decoder structure to obtain semantic segmentation graph, in downsampling we use dilated convolution to increase feature map size. And we also use label smoothing and adjust the loss function to improve the ability to predict the test set more accurately. Experiments show that this method is effective and has a good generalization for the detection problem of joints steering gear grooves.

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

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

[3]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[4]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[5]  Mahesh Pal,et al.  Random forest classifier for remote sensing classification , 2005 .

[6]  Yi Yang,et al.  Random Erasing Data Augmentation , 2017, AAAI.

[7]  George Papandreou,et al.  Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation , 2018, ECCV.

[8]  Geoffrey E. Hinton,et al.  When Does Label Smoothing Help? , 2019, NeurIPS.

[9]  Roberto Cipolla,et al.  Semantic texton forests for image categorization and segmentation , 2008, 2008 IEEE Conference on Computer Vision and Pattern Recognition.

[10]  Kaiming He,et al.  Feature Pyramid Networks for Object Detection , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[11]  Roberto Cipolla,et al.  SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[12]  Zenghui Wang,et al.  Deep Convolutional Neural Networks for Image Classification: A Comprehensive Review , 2017, Neural Computation.

[13]  Trevor Darrell,et al.  Fully Convolutional Networks for Semantic Segmentation , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[14]  Seong Joon Oh,et al.  CutMix: Regularization Strategy to Train Strong Classifiers With Localizable Features , 2019, 2019 IEEE/CVF International Conference on Computer Vision (ICCV).

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