基于三维卷积神经网络的视网膜OCT图像的分类方法
基本信息
申请号 | CN201710506132.5 | 申请日 | - |
公开(公告)号 | CN107437092B | 公开(公告)日 | 2019-11-15 |
申请公布号 | CN107437092B | 申请公布日 | 2019-11-15 |
分类号 | G06K9/62;G06N3/04 | 分类 | 计算;推算;计数; |
发明人 | 陈新建;刘云 | 申请(专利权)人 | 苏州比格威医疗科技有限公司 |
代理机构 | 南京纵横知识产权代理有限公司 | 代理人 | 苏州比格威医疗科技有限公司 |
地址 | 215011江苏省苏州市高新区竹园路209号 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种基于三维卷积神经网络的视网膜OCT图像的分类算法,包括以下步骤:S01:采集三种视网膜OCT图像,对三种视网膜进行分类标记;S02:数据预处理,对三维OCT图像数据进行降采样,得到统一大小的三维图像用于输入三维卷积神经网络;S03:根据迁移学习思想,用大量标记好的自然图像预训练一个三维卷积神经网络模型;S04:用预处理好的视网膜OCT图像对训练好的三维卷积神经网络模型进行微调,在主流网络中间的卷积层后加入分支网络,将主流网络和所述分支网络的输出层进行融合;S05:将测试图像按照S02步骤进行预处理,利用S04中微调后的所述三维卷积神经网络模型进行测试,输出分类结果。本发明具有可以对三维视网膜OCT图像进行分类以及提高分类准确性的特点。 |
