基于SimAM-YOLOv5s的PCB缺陷檢測算法
無線電工程
頁數: 10 2023-09-07
摘要: 為解決PCB缺陷檢測存在精度低、檢測效果差的問題,提出一種基于SimAM-YOLOv5s的PCB缺陷檢測算法。利用Kmeans++對錨框進行重新聚類,通過添加淺層尺度信息來豐富小目標數據,提高深層和淺層語義信息的融合能力;將損失函數修改為SIoU,即引入角度損失來計算距離損失,以加快網絡收斂速度,使回歸參數更加準確;通過結合輕量化注意力機制SimAM為網絡提供三維的注意力權重,... (共10頁)