融合SA注意力机制的YOLO5s在石油油管表面缺陷检测的应用

摘要： 针对石油厂油管表面缺陷检测存在检测精度低、速度慢和模型复杂等问题，提出一种SA-YOLO算法。以YOLOv5s模型为基础，对原数据集进行预处理，采用BoTNet Transformer结构代替Backbone特征主干的部分卷积，并用multi-head self-attention(MHSA)替换卷积层，以减少网络层，同时提高获取全局信息的能力；最后，将Shuffle Atte... （共8页）