Abstract:Aluminum Profiles (APs) are aluminum materials obtained by hot melting and extruding aluminum rods. It has the characteristics of low cost, strong plasticity, easy processing, and recyclability, and therefore plays an important role in industrial production. However, defects such as Non-Conductive (NC), Scratch, Orange Peel (OP), and Dirty Point (DP) often occur during the production and processing of APs, which can seriously affect the quality of APs. In addition, surface defects of APs also have problems such as fuzzy regional definition, large-scale variation, imbalance of aspect ratio, and high inter-class defect similarity, making defect detection more challenging. To solve these problems, this paper proposes an attention-guided object detection algorithm called MA-YOLO, specifically for Surface Defect Detection (SDD) of APs. The algorithm is based on YOLOv5s. Firstly, the K-Means++ clustering algorithm is used to optimize the anchor boxes, which alleviates the problem of aspect ratio imbalance. Secondly, by improving the multi-scale Feature Fusion Network (FFN), the detection performance of the model to detect the defects with unbalanced aspect ratio is improved, and the adaptability of the model to defects of different scales is enhanced. Finally, a novel Max Pooling Average Pooling (MA) attention module is proposed to improve the overall detection performance of the model, especially for small-scale defects. Experimental results on the aluminum profile surface defect dataset show that MA-YOLO has better detection performance and superiority than the current mainstream object detection algorithms, and compared with the baseline YOLOv5s, the mAP 50 and F1 score are increased by 2.9% and 2.2%, respectively, while keeping the model lightweight. This indicates that MA-YOLO has broad application prospects in the surface defect detection of APs.

ADDet: An Efficient Multiscale Perceptual Enhancement Network for Aluminum Defect Detection

An efficient multiscale enhancement network with attention mechanism for aluminium defect detection

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

An Algorithm for Real-Time Aluminum Profile Surface Defects Detection Based on Lightweight Network Structure

Aluminum surface defect detection method based on a lightweight YOLOv4 network

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

Aluminum Product Surface Defect Detection Method Based on Improved CenterNet

Surface Defect Detection for Aerospace Aluminum Profiles with Attention Mechanism and Multi-Scale Features

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

Improved YOLOv5 Surface Defect Detection Algorithm for Aluminum

AEGLR-Net: Attention Enhanced Global-Local Refined Network for Accurate Detection of Car Body Surface Defects

Efficient detection of multiscale defects on metal surfaces with improved YOLOv5

Surface defect detection method of aluminium alloy castings based on data enhancement and CRT‐DETR

Deformable Feature Pyramid Network for Aluminum Profile Surface Defect Detection

A defect detection method for industrial aluminum sheet surface based on improved YOLOv8 algorithm

An Improved Feature Enhancement CenterNet Model for Small Object Defect Detection on Metal Surfaces

Unleashing the power of AI in detecting metal surface defects: an optimized YOLOv7-tiny model approach

ALdamage-seg: A Lightweight Model for Instance Segmentation of Aluminum Profiles

MA-YOLO: A Method for Detecting Surface Defects of Aluminum Profiles With Attention Guidance

DAssd-Net: A Lightweight Steel Surface Defect Detection Model Based on Multi-Branch Dilated Convolution Aggregation and Multi-Domain Perception Detection Head

SPEED:Semantic Prior and Extremely Efficient Dilated Convolution Network for Real-Time Metal Surface Defects Detection