Recursive Residual Convolutional Neural Network with Attention Mechanism Based on U-Net (Attention R2U-Net) for Surface Minor Defect Detection
摘要
Surface defect detection is a critical task in industrial production, involving the identification of defects in material surface images through advanced algorithms. While existing segmentation-based approaches struggle with issues such as weak anti-interference, susceptibility to noise, and poor performance in detecting minor defects, this paper introduces an enhanced approach using the R2U-Net framework combined with an attention mechanism. Our model integrates recurrent convolutional layers, residual connections, and attention gates within the U-Net framework to improve feature extraction, gradient flow, and focus on relevant regions. These enhancements allow for more accurate detection of small defects by iteratively refining features, improving network depth, and dynamically suppressing background noise. Compared to existing models, our approach demonstrates superior accuracy and precision in detecting small surface defects, making it a valuable contribution to industrial computer vision applications.