Lightweight DCGAN and MobileNet based model for detecting X-ray welding defects under unbalanced samples

基于轻量级 DCGAN 和 MobileNet 的模型用于检测不平衡样本下的 X 射线焊接缺陷

阅读:1

Abstract

X-ray nondestructive testing technology is widely used for inspecting welds and identifying defects, and is crucial in the manufacturing industry. However, the diversity of welding defects and the imbalanced defect samples reduce defect classification model accuracy and can cause classifier overfitting. This paper proposes an improved DCGAN model for generating welding defect samples by integrating deep convolutional neural networks to enhance the training relationship between the generator and discriminator, thus increasing the number of training samples. We introduce a lightweight DG-MobileNet model to address low accuracy in welding defect identification and poor model convergence. Dilated convolution modules and Squeeze-and-Excitation self-attention mechanisms expand the convolutional receptive field and improve feature extraction capabilities. The fully connected layer is replaced by a global average pooling layer, reducing training parameters and mitigating overfitting. Additionally, combining DropBlock technology with Batch Normalization optimizes the feature extraction process, enhancing generalization ability. The experimental results demonstrate that the proposed model achieves a recognition accuracy of 98.78% and exhibits superior performance in terms of efficiency and lightweight design, highlighting its potential for deployment in real-world industrial applications.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。