一種SSD目標(biāo)檢測(cè)卷積神經(jīng)網(wǎng)絡(luò)的加速計(jì)算方法

基本信息

申請(qǐng)?zhí)?/td> CN201910846935.4 申請(qǐng)日 -
公開（公告）號(hào) CN110647988A 公開（公告）日 2020-01-03
申請(qǐng)公布號(hào) CN110647988A 申請(qǐng)公布日 2020-01-03
分類號(hào) G06N3/08（2006.01）; G06N3/04（2006.01） 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 韓夢(mèng)江; 樓燚航; 白燕; 張永祥; 陳杰 申請(qǐng)（專利權(quán)）人 博云視覺（北京）科技有限公司
代理機(jī)構(gòu) 北京眾澤信達(dá)知識(shí)產(chǎn)權(quán)代理事務(wù)所（普通合伙） 代理人 博云視覺（北京）科技有限公司
地址 100193 北京市海淀區(qū)西北旺東路10號(hào)院東區(qū)2號(hào)樓3層303A室
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種SSD目標(biāo)檢測(cè)卷積神經(jīng)網(wǎng)絡(luò)的加速計(jì)算方法，包括以下步驟：S1、將SSD目標(biāo)檢測(cè)卷積神經(jīng)網(wǎng)絡(luò)訓(xùn)練階段的目標(biāo)框Box回歸層和目標(biāo)分類回歸層刪除，使用一個(gè)新的卷積層同時(shí)進(jìn)行Box推導(dǎo)計(jì)算和目標(biāo)分類計(jì)算；S2、在新的卷積層后添加一個(gè)BlobSplit層，通過BlobSplit層將新的卷積層輸出的Blob按照配置的channel拆分成兩個(gè)獨(dú)立的Blob并輸出給訓(xùn)練階段后續(xù)的兩個(gè)相關(guān)層；S3、將SSD目標(biāo)檢測(cè)卷積神經(jīng)網(wǎng)絡(luò)推導(dǎo)階段的目標(biāo)框Box回歸層和目標(biāo)分類回歸層刪除，使用一個(gè)新的卷積層同時(shí)進(jìn)行Box推導(dǎo)計(jì)算和目標(biāo)分類計(jì)算；S4、將SSD目標(biāo)檢測(cè)卷積神經(jīng)網(wǎng)絡(luò)推導(dǎo)階段的后續(xù)相關(guān)層全部刪除，在新的卷積層后連接一個(gè)Object_Detector層。