Deep learning-based weld defect classification using VGG16 transfer learning adaptive fine-tuning

基于深度学习的VGG16迁移学习自适应微调焊接缺陷分类

阅读:1

Abstract

Welding is a vital joining process; however, occurrences of weld defects often degrade the quality of the welded part. The risk of occurrence of a variety of defects has led to the development of advanced weld defects detection systems such as automated weld defects detection and classification. The present work is a novel approach that proposes and investigates a unique image-centered method based on a deep learning model trained by a small X-ray image dataset. A data augmentation method able to process images on the go was used to offset the limitation of the small X-ray dataset. Fine-tuned transfer learning techniques were used to train two convolutional neural network based architectures with VGG16 and ResNet50 as the base models for the augmented sets. Out of the networks we fine-tuned, VGG16 based model performed well with a relatively higher average accuracy of 90%. Even though the small dataset was spread across 15 different classes in an unbalanced way, the learning curves showed acceptable model generalization characteristics.

特别声明

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

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

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

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