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