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