融合注意力机制的铝型材缺陷检测研究OA
Defect detection of aluminum profiles with attention mechanism incorporation
针对现有检测模型在小目标缺陷识别中存在漏检、误检等问题,提出一种基于YOLOv8n-SE模型的铝型材表面缺陷检测方法.通过在YOLOv8n模型的颈部网络中嵌入SE注意力机制,增强特征提取能力,提高对缺陷区域的定位精度与检测敏感度.基于铝型材工件缺陷数据集开展对比试验,将改进模型与Faster R-CNN、YOLOv5n等轻量级模型及不同注意力机制的方案进行性能评估.研究结果表明,改进后模型的平均精确度(mAP)达到75.0%,相较于原始YOLOv8n模型提升4.2%,参数量基本保持不变,推理速度仅下降0.3%.嵌入SE注意力机制的YOLOv8n模型能有效提升铝型材表面缺陷识别效果,缓解小目标的漏检与误检问题,同时保持轻量级模型的高效推理优势,适用于工业场景下的铝型材缺陷检测需求.研究结论为同类工业缺陷检测任务的模型选择与优化提供参考.
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.
李峰
中国飞行试验研究院 中飞测试,陕西 西安 710089
信息技术与安全科学
缺陷检测铝型材YOLOv8模型SE注意力机制小目标检测
defect detectionaluminum profilesYOLOv8 modelSE attention mechanismsmall target detection
《辽宁工程技术大学学报(自然科学版)》 2026 (2)
249-256,8
陕西省重点研发计划项目(2023JBGS-23)
评论