特異小樣本工業(yè)產(chǎn)品表面缺陷檢測方法研究
河南師范大學(xué)學(xué)報(bào)(自然科學(xué)版)
頁數(shù): 9 2024-07-01
摘要: 基于機(jī)器視覺的工業(yè)產(chǎn)品表面缺陷檢測設(shè)備和系統(tǒng)大量應(yīng)用在工業(yè)制造領(lǐng)域,目前其難點(diǎn)在于工業(yè)檢測數(shù)據(jù)的采集,由于訓(xùn)練樣本缺失導(dǎo)致深度學(xué)習(xí)網(wǎng)絡(luò)模型無法有效訓(xùn)練.為解決上述問題,首先,提出一種基于不規(guī)則掩碼的傷痕樣本生成算法,改善了鋼板表面缺陷檢測任務(wù)中特異小樣本數(shù)據(jù)集正負(fù)樣本不均衡的情況;然后,在YOLOv8主干網(wǎng)絡(luò)引入MHSA多頭自注意力,提高對鋼板表面缺陷的關(guān)注度;最后,使用SIo... (共9頁)