一種無錨點(diǎn)框的三維肺結(jié)節(jié)檢測模型訓(xùn)練方法及裝置
基本信息
| 申請(qǐng)?zhí)?/td> | CN202110248609.0 | 申請(qǐng)日 | - |
| 公開（公告）號(hào) | CN112614133B | 公開（公告）日 | 2021-07-06 |
| 申請(qǐng)公布號(hào) | CN112614133B | 申請(qǐng)公布日 | 2021-07-06 |
| 分類號(hào) | G06T7/00 | 分類 | 計(jì)算；推算；計(jì)數(shù); |
| 發(fā)明人 | 杜強(qiáng);王思其;郭雨晨;聶方興;唐超 | 申請(qǐng)（專利權(quán)）人 | 北京小白世紀(jì)網(wǎng)絡(luò)科技有限公司 |
| 代理機(jī)構(gòu) | 北京八月瓜知識(shí)產(chǎn)權(quán)代理有限公司 | 代理人 | 秦瑩 |
| 地址 | 100083 北京市海淀區(qū)王莊路1號(hào)院清華同方科技大廈D座19層1903-1號(hào) | ||
| 法律狀態(tài) | - | ||
摘要
| 摘要 | 本發(fā)明公開了一種無錨點(diǎn)框的三維肺結(jié)節(jié)檢測模型訓(xùn)練方法及裝置，該方法包括：獲取數(shù)據(jù)集中三維肺結(jié)節(jié)樣本的圖像數(shù)據(jù)、標(biāo)注類別和標(biāo)注坐標(biāo)點(diǎn)并構(gòu)建三維肺結(jié)節(jié)數(shù)據(jù)庫；將進(jìn)行數(shù)據(jù)增強(qiáng)后的圖像數(shù)據(jù)輸入到卷積神經(jīng)網(wǎng)絡(luò)進(jìn)行處理，得到多個(gè)特征圖；將特征圖輸入到預(yù)先設(shè)置的具有兩個(gè)卷積分支的共享頭部網(wǎng)絡(luò)，輸出預(yù)測類別、預(yù)測中心度和預(yù)測坐標(biāo)點(diǎn)，計(jì)算預(yù)測置信度和標(biāo)注中心度，將標(biāo)注坐標(biāo)點(diǎn)、預(yù)測坐標(biāo)點(diǎn)、預(yù)測置信度、標(biāo)注類別、預(yù)測中心度、標(biāo)注中心度輸入到損失函數(shù)，反向傳播，訓(xùn)練模型。采用本發(fā)明實(shí)施例，解決了有錨點(diǎn)框所帶來的問題、減少了很多的計(jì)算量，采用雙分支的網(wǎng)絡(luò)結(jié)構(gòu)使得不同任務(wù)間的特征不互相干擾，模型結(jié)果更加準(zhǔn)確。 |
