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.

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

Improved YOLOv5 Surface Defect Detection Algorithm for Aluminum

Detection Algorithm for Bearing Roller End Surface Defects Based on Improved YOLOv5n and Image Fusion

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

Aluminum surface defect detection method based on a lightweight YOLOv4 network

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

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

Efficient detection of multiscale defects on metal surfaces with improved YOLOv5

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

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

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

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

Automatic Defect Recognition Method of Aluminium Profile Surface Defects

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

Fast and Accurate Defects Detection for Additive Manufactured Parts by Multispectrum and Machine Learning

YOLOv5-ACCOF Steel Surface Defect Detection Algorithm

CFM-YOLOv5:CFPNet moudle and muti-target prediction head incorporating YOLOv5 for metal surface defect detection

Object Detection Algorithm for Surface Defects Based on a Novel YOLOv3 Model

A Feature-Oriented Reconstruction Method for Surface-Defect Detection on Aluminum Profiles

Deformable Feature Pyramid Network for Aluminum Profile Surface Defect Detection

LCG-YOLO: A Real-Time Surface Defect Detection Method for Metal Components