文件大小：1.17M
摘要：YOLOv8n在处理热轧钢材表面缺陷时，存在复杂特征提取效果欠佳、定位精度差、漏检率高的问题。因此，本文提出一种改进YOLOv8n的热轧钢材表面缺陷检测模型GSX-YOLOv8n。首先，为了强化模型的特征提取能力，提高模型的召回率，基于空间注意力和通道注意力，设计了全局卷积注意力模块。其次，在主干网络中使用SPD-Conv卷积，提高模型的低分辨率图像检测能力。最后，在网络结构中添加小目标检测头，利用上下文关系，提高模型的小目标检测检测能力。实验结果显示，改进过后的模型mAP@0.5达到82.2%，mAP@0.5:0.95达到49.1%，召回率提高到79.0%，表明本文模型对热轧钢材表面缺陷具有更好的检测性能。
文章目录
1 YOLOv8算法
1.1 头部（Head）
1.2 骨干（Backbone）
1.3 颈部（Neck）
2 GSX-YOLOv8n模型
2.1 GCA注意力模块
2.2 SPD-Conv
2.3 小目标检测层
2.4 方法总结
3 实验
3.1 数据集
3.2 评估指标
3.3 实验设计
3.3.1 多种注意力机制
3.3.2 消融实验
3.3.3 对比实验结果
3.3.4 检测结果可视化对比
4 结论
