基于改進(jìn)YOLOv7的鋼軌缺陷檢測方法
摘要: 針對鐵路軌道缺陷檢測精度低,漏檢率高,實時性不足的問題,本文提出了一種基于YOLO-FCA的鋼軌缺陷檢測算法。首先,將YOLOv7的主干網(wǎng)絡(luò)替換成FasterNet輕量網(wǎng)絡(luò),并加入CloAttention注意力模塊,減少參數(shù)量和計算負(fù)載的同時提高缺陷檢測的精度。其次,提出MS-ASFF,獲取高層語義信息和保留低層詳細(xì)特征,增強(qiáng)模型檢測的準(zhǔn)確性和魯棒性。最后,在不影響精度的情況下... (共9頁)
開通會員,享受整站包年服務(wù)