基于SDD-YOLO 的輕量級帶鋼缺陷實時檢測算法

打開文本圖片集

摘 要： 針對復雜工業(yè)生產(chǎn)環(huán)境下，熱軋帶鋼的表面缺陷檢測準確率低，網(wǎng)絡模型參數(shù)量過大，難以部署等問題，該文提出一種基于YOLOv4 模型改進的輕量級帶鋼缺陷實時檢測算法SDD-YOLO。所提算法在特征提取部分采用GhostNet 網(wǎng)絡，壓縮模型參數(shù)量；在特征融合部分，借鑒BiFPN 結(jié)構(gòu)改進PAN 網(wǎng)絡，采用GSConv 卷積代替標準卷積，減少模型參數(shù)量和計算量，同時嵌入注意力模塊CA（Coordinate Attention），增強模型特征融合能力；在預測部分采用SIOU-loss 代替CIOU-loss，提高模型收斂效率，加快收斂速度；采用k-means 聚類算法重新設計先驗框，提高模型精度。（剩余1038字）