一種基于多點位雙分支模型的工業(yè)視覺缺陷檢測方法

基本信息

申請?zhí)?/td> CN202111055665.9 申請日 -
公開(公告)號 CN113935947A 公開(公告)日 2022-01-14
申請公布號 CN113935947A 申請公布日 2022-01-14
分類號 G06T7/00(2017.01)I;G06K9/62(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I;G06V10/764(2022.01)I;G06V10/80(2022.01)I;G06V10/82(2022.01)I 分類 計算;推算;計數(shù);
發(fā)明人 宋艷枝;汪方軍 申請(專利權(quán))人 合肥高斯智能科技有限公司
代理機構(gòu) 合肥洪雷知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 代理人 郎海云
地址 230000安徽省合肥市蜀山區(qū)金寨路91號立基大廈A座第16層1603室
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于多點位雙分支模型的工業(yè)視覺缺陷檢測方法,涉及人工智能技術(shù)領(lǐng)域。本發(fā)明包括如下步驟:根據(jù)點位劃分?jǐn)?shù)據(jù):根據(jù)點位信息對數(shù)據(jù)進行劃分,得到多個正樣本池和負(fù)樣本池,并進行樣本對齊;多點位采樣策略:根據(jù)多點位采樣策略分別從正樣本池和負(fù)樣本池中挑選正樣本對和負(fù)樣本對;特征提取:以ResNet?50模型作為特征提取器,進行輸入數(shù)據(jù)的特征提取;雙分支預(yù)測:將提取的特征輸入到包含注意力機制的若干全連接層,進行缺陷預(yù)測。本發(fā)明通過利用點位信息,可以有效地劃分?jǐn)?shù)據(jù)集,將復(fù)雜多樣的數(shù)據(jù)集轉(zhuǎn)換為若干點位內(nèi)的子數(shù)據(jù)集,從而降低了網(wǎng)絡(luò)學(xué)習(xí)難度,提升了模型的缺陷識別能力。