DSNet: A Dynamic Squeeze Network for Real-time Weld Seam Image Segmentation - Archive ouverte HAL Accéder directement au contenu
Article Dans Une Revue Engineering Applications of Artificial Intelligence Année : 2024

DSNet: A Dynamic Squeeze Network for Real-time Weld Seam Image Segmentation

Résumé

The image noise generated by the welding process, such as arc light, splash, and smoke, brings significantchallenges for the laser vision sensor-based welding robot to locate the weld seam and accurately conduct automatic welding. Currently, deep learning-based approaches surpass traditional methods in flexibility and robustness. However, their significant computational cost leads to a mismatch with the real-time requirement of automated welding. In this paper, we propose an efficient hybrid architecture of Convolutional Neural Network (CNN) and transformer, referred to as Dynamic Squeeze Network (DSNet), for real-time weld seam segmentation. More precisely, a lightweight segmentation framework is developed to fully leverage the advantages of the transformer structure without significantly increasing computational overhead. In this respect, an efficient encoder, which aims to increase its features diversity, has been designed and resulted in substantial improvement of encoding performance. Moreover, we propose a plug-and-play lightweight attention module that generates more effective attention weights by exploiting statistical information of weld seam data and introducing linear priors. Extensive experiments on weld seam images using NVIDIA GTX 1050Ti show that our approach reduces the number of parameters by 54x, decreases computational complexity by 34x, and improves inference speed by 33x compared to the baseline method TransUNet. DSNet achieves superior accuracy (78.01% IoU, 87.64% Dice) and speed performance (100 FPS) with lower model complexity and computational burden than most state-of-the-art methods. The code is available at https://github.com/hackerschen/DSNet.
Fichier principal
Vignette du fichier
DSN_Revised_accepted.pdf (3.71 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04505581 , version 1 (18-03-2024)

Identifiants

  • HAL Id : hal-04505581 , version 1

Citer

Jia Chen, Congcong Wang, Fan Shi, Mounir Kaaniche, Meng Zhao, et al.. DSNet: A Dynamic Squeeze Network for Real-time Weld Seam Image Segmentation. Engineering Applications of Artificial Intelligence, 2024. ⟨hal-04505581⟩
0 Consultations
4 Téléchargements

Partager

Gmail Facebook X LinkedIn More