基于三維卷積神經(jīng)網(wǎng)絡(luò)的視網(wǎng)膜OCT圖像的分類方法

基本信息

申請?zhí)?/td> CN201710506132.5 申請日 -
公開(公告)號 CN107437092B 公開(公告)日 2019-11-15
申請公布號 CN107437092B 申請公布日 2019-11-15
分類號 G06K9/62;G06N3/04 分類 計算;推算;計數(shù);
發(fā)明人 陳新建;劉云 申請(專利權(quán))人 蘇州比格威醫(yī)療科技有限公司
代理機(jī)構(gòu) 南京縱橫知識產(chǎn)權(quán)代理有限公司 代理人 蘇州比格威醫(yī)療科技有限公司
地址 215011江蘇省蘇州市高新區(qū)竹園路209號
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于三維卷積神經(jīng)網(wǎng)絡(luò)的視網(wǎng)膜OCT圖像的分類算法,包括以下步驟:S01:采集三種視網(wǎng)膜OCT圖像,對三種視網(wǎng)膜進(jìn)行分類標(biāo)記;S02:數(shù)據(jù)預(yù)處理,對三維OCT圖像數(shù)據(jù)進(jìn)行降采樣,得到統(tǒng)一大小的三維圖像用于輸入三維卷積神經(jīng)網(wǎng)絡(luò);S03:根據(jù)遷移學(xué)習(xí)思想,用大量標(biāo)記好的自然圖像預(yù)訓(xùn)練一個三維卷積神經(jīng)網(wǎng)絡(luò)模型;S04:用預(yù)處理好的視網(wǎng)膜OCT圖像對訓(xùn)練好的三維卷積神經(jīng)網(wǎng)絡(luò)模型進(jìn)行微調(diào),在主流網(wǎng)絡(luò)中間的卷積層后加入分支網(wǎng)絡(luò),將主流網(wǎng)絡(luò)和所述分支網(wǎng)絡(luò)的輸出層進(jìn)行融合;S05:將測試圖像按照S02步驟進(jìn)行預(yù)處理,利用S04中微調(diào)后的所述三維卷積神經(jīng)網(wǎng)絡(luò)模型進(jìn)行測試,輸出分類結(jié)果。本發(fā)明具有可以對三維視網(wǎng)膜OCT圖像進(jìn)行分類以及提高分類準(zhǔn)確性的特點。