基于改進(jìn)YOLOv7算法的連鑄坯表面缺陷檢測

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