Surface Defect Detection with Gabor Filter Using Reconstruction-Based Blurring U-Net-ViT
This addresses the problem of detecting defects in industrial surfaces for quality control, but it is incremental as it builds on existing U-Net and ViT methods with added filters and noise handling.
The paper tackled surface defect detection by combining Gabor filters with a blurring U-Net-ViT model to improve accuracy and reliability across various textures, achieving an average AUC of 0.939 on datasets like MVTec-AD.
This paper proposes a novel approach to enhance the accuracy and reliability of texture-based surface defect detection using Gabor filters and a blurring U-Net-ViT model. By combining the local feature training of U-Net with the global processing of the Vision Transformer(ViT), the model effectively detects defects across various textures. A Gaussian filter-based loss function removes background noise and highlights defect patterns, while Salt-and-Pepper(SP) masking in the training process reinforces texture-defect boundaries, ensuring robust performance in noisy environments. Gabor filters are applied in post-processing to emphasize defect orientation and frequency characteristics. Parameter optimization, including filter size, sigma, wavelength, gamma, and orientation, maximizes performance across datasets like MVTec-AD, Surface Crack Detection, and Marble Surface Anomaly Dataset, achieving an average Area Under the Curve(AUC) of 0.939. The ablation studies validate that the optimal filter size and noise probability significantly enhance defect detection performance.