Journal
APPLIED SCIENCES-BASEL
Volume 12, Issue 13, Pages -Publisher
MDPI
DOI: 10.3390/app12136569
Keywords
defect detection; deep convolution generation adversarial network; data augmentation; lightweight model
Categories
Funding
- Key Research and Development Program of Zhejiang Province [2022C01139, 2019C01134]
- Key Research and Development Program of Jinhua [2021-1-001a]
Ask authors/readers for more resources
This paper proposes a small sample gear face defect detection method based on DCGAN and lightweight CNN. By data augmentation and defect classification, it achieves efficient online defect detection.
Online defect detection in small industrial parts is of paramount importance for building closed loop intelligent manufacturing systems. However, high-efficiency and high-precision detection of surface defects in these manufacturing systems is a difficult task and poses a major research challenge. The small sample size of industrial parts available for training machine learning algorithms and the low accuracy of computer vision-based inspection algorithms are the bottlenecks that restrict the development of efficient online defect detection technology. To address these issues, we propose a small sample gear face defect detection method based on a Deep Convolutional Generative Adversarial Network (DCGAN) and a lightweight Convolutional Neural Network (CNN) in this paper. Initially, we perform data augmentation by using DCGAN and traditional data enhancement methods which effectively increase the size of the training data. In the next stage, we perform defect classification by using a lightweight CNN model which is based on the state-of-the-art Vgg11 network. We introduce the Leaky ReLU activation function and a dropout layer in the proposed CNN. In the experimental evaluation, the proposed framework achieves a high score of 98.40%, which is better than that of the classic Vgg11 network model. The method proposed in this paper is helpful for the detection of defects in industrial parts when the available sample size for training is small.
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