一種基于深度學(xué)習(xí)的金屬零件快速分割方法

基本信息

申請(qǐng)?zhí)?/td> CN202210157462.9 申請(qǐng)日 -
公開(公告)號(hào) CN114494272A 公開(公告)日 2022-05-13
申請(qǐng)公布號(hào) CN114494272A 申請(qǐng)公布日 2022-05-13
分類號(hào) G06T7/10(2017.01)I;G06T7/00(2017.01)I;G06T7/70(2017.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分類 計(jì)算;推算;計(jì)數(shù);
發(fā)明人 李威燃;唐廷浩;夏齊平 申請(qǐng)(專利權(quán))人 蘇州才炬智能科技有限公司
代理機(jī)構(gòu) - 代理人 -
地址 215300江蘇省蘇州市昆山市花橋鎮(zhèn)徐公橋路2號(hào)中茵廣場(chǎng)E區(qū)705室
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于深度學(xué)習(xí)的金屬零件快速分割方法,包括:通過Labelme圖像標(biāo)注工具建立金屬零件數(shù)據(jù)集;輸入單張圖像,使用基于FPN網(wǎng)絡(luò)的物體檢測(cè)算法預(yù)測(cè)感興趣目標(biāo)區(qū)域和候選框坐標(biāo)(x,y,w,h),采用Resnet?50及特征金字塔作為基礎(chǔ)網(wǎng)絡(luò)來獲取整張輸入圖像的特征;將感興趣目標(biāo)區(qū)域建模為另兩個(gè)重疊圖層,分別為Top Layer和Bottom Layer,并進(jìn)行卷積操作以進(jìn)一步提取特征;使用RoI Align算法根據(jù)物體檢測(cè)框位置,在整張圖片特征圖內(nèi)準(zhǔn)確摳取感興趣目標(biāo)區(qū)域的特征子圖,將其作為雙圖卷積神經(jīng)網(wǎng)絡(luò)的輸入并用于最終的物體分割;根據(jù)自定義的損失函數(shù),更新模型參數(shù),迭代上述步驟,計(jì)算是否達(dá)到預(yù)設(shè)精度。