关键词:
铝型材
表面缺陷检测
注意力机制
YOLOv8
模型轻量化
GhostBottleneck
摘要:
在铝型材制造过程中,铝型材表面会因受到材料或加工工艺等因素的影响而产生擦花、脏点等缺陷,直接影响铝型材使用性能。分析了铝型材表面缺陷特点并对比现有深度学习目标检测算法,基于YOLOv8网络模型提出了一种融合GhostBottleneck及注意力机制的铝型材表面缺陷检测算法。首先,将Ghost卷积引入Bottleneck层,并用DWConv替换骨干网络中部分卷积结构,在保证检测精度的同时,降低模型复杂程度;然后,在此基础上,将注意力机制添加到YOLOv8检测头模块中,用于提高该模型的检测精度;最后,开展了实验验证,实验结果表明,融合GhostBottleneck及注意力机制的铝型材表面缺陷检测算法精度达到了0.932,相比基础YOLOv8算法,精度提升了5.9%,且模型运算参数量减少了24%,整体性能可满足工业上对铝型材缺陷检测的精度及速度要求。