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

基本信息

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

摘要

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