Abstract:The automatic classification of aluminum profile surface defects is of great significance in improving the surface quality of aluminum profiles in practical production. This classification is influenced by the small and unbalanced number of samples and lack of uniformity in the size and spatial distribution of aluminum profile surface defects. It is difficult to achieve high classification accuracy by directly using the current advanced classification algorithms. In this paper, digital image processing methods such as rotation, flipping, contrast, and luminance transformation were used to augment the number of samples and imitate the complex imaging environment in actual practice. A RepVGG with CBAM attention mechanism (RepVGG-CBAM) model was proposed and applied to classify ten types of aluminum profile surface defects. The classification accuracy reached 99.41%, in particular, the proposed method can perfectly classify six types of defects: concave line (cl), exposed bottom (eb), exposed corner bottom (ecb), mixed color (mc), non-conductivity (nc) and orange peel (op), with 100% precision, recall, and F1. Compared with the existing advanced classification algorithms VGG16, VGG19, ResNet34, ResNet50, ShuffleNet_v2, and basic RepVGG, our model is the best in terms of accuracy, macro precision, macro recall and macro F1, and the accuracy was improved by 4.85% over basic RepVGG. Finally, an ablation experiment proved that the classification ability was strongest when the CBAM attention mechanism was added following Stage 1 to Stage 4 of RepVGG. Overall, the method we proposed in this paper has a significant reference value for classifying aluminum profile surface defects.

Aluminum Plate Surface Defects Classification Based on the BP Neural Network

Recognition algorithm on bearing surface defect type

Study of Bp Network for A Cylinder Shell'S Support Identification

Deformable Feature Pyramid Network for Aluminum Profile Surface Defect Detection

Research on Recognition Technology of Aluminum Profile Surface Defects Based on Deep Learning

Research on image classification method of strip steel surface defects based on improved Bat algorithm optimized BP neural network

Defects Rapid Identification of Marking Region Based on Binarization and BP Neural Network

Automatic Defect Recognition Method of Aluminium Profile Surface Defects

Surface Defect Detection of Aluminum Profiles Based on Multiscale and Self-Attention Mechanisms

Strip Surface Defect Recognition Algorithm Based on PCA and Improved BP Neural Network

An Effective Surface Defect Classification Method Based on RepVGG with CBAM Attention Mechanism (RepVGG-CBAM) for Aluminum Profiles

Application of an improved sparrow search algorithm in BP network classification of strip steel surface defect images

Solar Cells Surface Defects Detection Based on Deep Learning

Paper Defects Classifier Design Based on BP Neural Network

Aluminum Product Surface Defect Detection Method Based on Improved CenterNet

Defect detection of printed circuit board based on adaptive key-points localization network

A Lightweight Deep Learning Network with an Optimized Attention Module for Aluminum Surface Defect Detection

An Improved MB-LBP Defect Recognition Approach for the Surface of Steel Plates

Approaches for Accuracy Improvement of the X-ray Image Defect Detection of Automobile Casting Aluminum Parts Based on Deep Learning

The Study on The Application of BP Neural Network Based on Genetic Algorithm Optimization in Magnetic Flux Leakage (MFL) of Rail Top Surface Stripping Defects

Defect Detection Method of Aluminum Profile Surface Using Deep Self-Attention Mechanism Under Hybrid Noise Conditions