融合SA注意力機制的YOLO5s在石油油管表面缺陷檢測的應用

摘要： 針對石油廠油管表面缺陷檢測存在檢測精度低、速度慢和模型復雜等問題，提出一種SA-YOLO算法。以YOLOv5s模型為基礎，對原數據集進行預處理，采用BoTNet Transformer結構代替Backbone特征主干的部分卷積，并用multi-head self-attention(MHSA)替換卷積層，以減少網絡層，同時提高獲取全局信息的能力；最后，將Shuffle Atte... （共8頁）