Journal
OPTICS AND LASERS IN ENGINEERING
Volume 117, Issue -, Pages 40-48Publisher
ELSEVIER SCI LTD
DOI: 10.1016/j.optlaseng.2019.01.011
Keywords
Surface inspection; Defect detection; Semi-supervised learning; Convolutional autoencoder; Generative adversarial networks
Categories
Funding
- National Natural Science Foundation of China [51674031]
- National Key R&D Program of China [2018YPB0704304]
Ask authors/readers for more resources
Defect inspection is extremely crucial to ensure the quality of steel surface. It affects not only the subsequent production, but also the quality of the end-products. However, due to the rare occurrence and appearance variations of defects, surface defect identification of steels has always been a challenging task. Recently, deep learning methods have shown outstanding performance in image classification, especially when there are enough training samples. Since most sample images of steel surface are unlabeled, a new semi-supervised learning method is proposed to classify surface defects of steels. The new method is named CAE-SGAN, as it is based on Convolutional Autoencoder (CAE) and semi-supervised Generative Adversarial Networks (SGAN). CAE-SGAN first trains a stacked CAE through massive unlabeled data. Considering the appearance variations of defects, the passthrough layer is used to help CAE extract fine-grained features. After CAE is trained, the encoder network of CAE is reserved as the feature extractor and fed into a softmax layer to form a new classifier. SGAN is introduced for semi-supervised learning to further improve the generalization ability of the new method. The classifier is trained with images collected from real production lines and images randomly generated by SGAN. Extensive experiments are carried out with samples captured from different steel production lines, and the results Indicate that CAE-SGAN had yielded best performances compared with traditional methods. Especially for hot rolled plates, the classification rate is improved by around 16%.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available