Welding defect detection with image processing on a custom small dataset: A comparative study

Abstract

This work focuses on detecting defects in welding seams using the most advanced You Only Look Once (YOLO) algorithms and transfer learning. To this end, the authors prepared a small dataset of images using manual welding and compared the performance of the YOLO v5, v6, v7, and v8 methods after two-step training. Key findings reveal that YOLOv7 demonstrates superior performance, suggesting its potential as a valuable tool in automated welding quality control. The authors’ research underscores the importance of model selection. It lays the groundwork for future exploration in larger datasets and varied welding scenarios, potentially contributing to defect detection practices in manufacturing industries. The dataset and the code repository links are also provided to support our findings.