Research on Weld Detection Based on Weighted Feature Fusion Network

Abstract

Weld seam grinding is a crucial process for welding cast products. Not only are the polished welds more attractive and durable, but they also have superior stress effects. However, the workplace is harsh and can significantly impact health. The first task for automated weld grinding is weld detection. This paper, a pyramid feature fusion network (YOLOv5-BiCA) learning model with embedded coordinate attention as a weighting condition that fully utilizes the shallow and deep network information. An online data augmentation strategy employing Mosaic+Mixup is proposed to address the problem of insufficient sample size. We use Focal Loss to enhance the classification function and increase the positive influence of positive samples on the loss function during the training process to address the problem of an imbalance between positive and negative sample. Our proposed YOLOv5-BiCA performs better in weld detection, mAP improves by 3.5%, and recall improves by 7.97%, according to experimental results based on homegrown datasets.