基于改進YOLOv7算法的連鑄坯表面缺陷檢測
組合機床與自動化加工技術
頁數: 5 2024-07-19
摘要: 針對連鑄坯表面缺陷檢測存在的參數量大、小目標檢測效果較差、人工標注數據集質量的不穩(wěn)定性等問題,提出了一種基于改進YOLOv7算法的連鑄坯表面缺陷檢測模型。首先,引入MobileNetv3 block模塊輕量化主干網絡,降低計算量,減少參數與復雜度;其次,融合雙層路由注意力機制,實現更靈活的計算分配和內容感知;最后,使用WIoU損失函數替換原損失函數,使模型獲得更好的泛化能力。改... (共5頁)