Welding Groove Edge Detection Method Using Lightweight Fusion Model Based on Transfer Learning

Abstract

Groove edge detection is the prerequisite for weld seam deviation identification. A welding groove edge detection method based on transfer learning is presented as a solution to the inaccuracy of the conventional image processing method for extracting the edge of the welding groove. DenseNet and MobileNetV2 are used as feature extractors for transfer learning. Dense-Mobile Net is constructed using the skip connections structure and depthwise separable convolution. The Dense-Mobile Net training procedure consists of two stages: pre-training and model fusion fine-tuning. Experiments demonstrate that the proposed model accurately detects groove edges in MAG welding images. Using MIG welding images and the Pascal VOC2012 dataset to evaluate the generalization ability of the model, the relevant indicators are greater than those of Support Vector Machine (SVM), Fully Convolutional Networks (FCN), and UNet. The average single-frame detection time of the proposed model is 0.14 s, which meets the requirements of industrial real-time performance.