辽宁工程技术大学学报（自然科学版）2026，Vol.45Issue(2)：249-256,8.DOI:10.11956/j.issn.1008-0562.20250459
融合注意力机制的铝型材缺陷检测研究
Defect detection of aluminum profiles with attention mechanism incorporation
摘要
Abstract
In the field of industrial material production and management,traditional detection methods are difficult to meet the diverse detection requirements for surface coating cracks and other defects of aluminum profiles.Moreover,existing detection models have problems of missed detection and false detection of small targets.This paper proposes a surface defect detection method for aluminum profiles based on YOLOv8n-SE.By embedding the SE attention mechanism in the neck network of the YOLOv8n model,the feature extraction ability is enhanced,improving the positioning accuracy and defect sensitivity of the defect area.Experiments were conducted using an aluminum profile defect dataset,and the proposed method was compared with lightweight models such as Faster R-CNN and YOLOv5n combined with different attention mechanisms.The research results show that the average precision(mAP)of the improved model reaches 75.0%,a 4.2%higher than the original YOLOv8n model,with the number of parameters remaining basically unchanged and the inference speed only decreasing by 0.3%.The improved YOLOv8n model with the embedded SE attention mechanism can effectively improve the recognition effect of surface defects of aluminum profiles,solve the problem of missed detection and false detection of small targets,and maintain the efficient inference advantage of lightweight models,making it suitable for the defect detection requirements of aluminum profiles in industrial scenarios.关键词
缺陷检测/铝型材/YOLOv8模型/SE注意力机制/小目标检测Key words
defect detection/aluminum profiles/YOLOv8 model/SE attention mechanism/small target detection分类
信息技术与安全科学引用本文复制引用
李峰..融合注意力机制的铝型材缺陷检测研究[J].辽宁工程技术大学学报（自然科学版）,2026,45(2):249-256,8.基金项目
陕西省重点研发计划项目(2023JBGS-23) （2023JBGS-23）
