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

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