基于YOLO-GR算法的輕量化鋼材表面缺陷檢測(cè)
組合機(jī)床與自動(dòng)化加工技術(shù)
頁(yè)數(shù): 6 2023-11-20
摘要: 針對(duì)工業(yè)帶鋼缺陷檢測(cè)效率低、精度差、模型部署困難等技術(shù)不足,提出了一種改進(jìn)YOLOv5s算法的輕量化鋼材表面缺陷檢測(cè)模型:YOLO-GR模型。首先,通過引入GhostV2 Bottleneck輕量化模塊作為主干特征提取網(wǎng)絡(luò),以減少網(wǎng)絡(luò)的參數(shù)量,同時(shí)將特征融合網(wǎng)絡(luò)中的普通卷積塊替換為深度可分離卷積塊,進(jìn)一步降低模型的計(jì)算復(fù)雜度以優(yōu)化特征提取網(wǎng)絡(luò);然后,在檢測(cè)頭部分加入RepLK大...