The invention discloses a classification method of retinal OCT image 3D convolution based on neural network, which comprises the following steps: S01: three OCT retinal image acquisition, classification mark of three kinds of retina; S02: data preprocessing, 3D OCT image data were subsampled and obtain a three-dimensional image of the same size for 3D input neural network convolution; S03: according to the migration theory, with a large number of labeled natural image pre training a 3D convolution neural network model; S04: OCT image of three-dimensional retinal volume the trained neural network model with product fine-tuning pre, in the middle of the mainstream network layer convolution branch network will join. The output layer of the main network and the branch network integration; S05: the test of image preprocessing in accordance with S02 steps, the use of S04 in the fine-tuning The three dimensional convolution neural network model is used to test and output the classification results. The present invention has the characteristics of classifying the three-dimensional OCT image of the retina and improving the accuracy of the classification.
【技术实现步骤摘要】
基于三维卷积神经网络的视网膜OCT图像的分类算法
本专利技术涉及一种基于三维卷积神经网络的视网膜OCT图像的分类算法,属于视网膜图像分类
技术介绍
现有的视网膜自动分类技术大部分基于眼底彩照或者小视野的视网膜OCT,即以黄斑为中心或者以视神经乳头为中心的视网膜OCT图像。相比较眼底彩照,OCT图像具有无创、高速、高分辨率、三维成像等优点,不过OCT图像中视网膜分类目前仍面临诸多挑战:图像各类之间差异不明显,图像本身存在大量斑噪声等。这些问题使得传统的方法很难取得较为精确的分类效果。卷积神经网络具有强大的学习能力,其在医学图像分类(例如,乳腺X线肿瘤图像的分类、CT肺间质图像的分类、糖尿病性视网膜眼底彩照分类等)中已取得了巨大的成功。所以考虑将该框架用于视网膜OCT图像分类的任务中。然而,以上的医学图像都是二维图像,都使用二维卷积神经网络来达到分类效果,由于OCT图像为三维图像,直接使用传统的二维卷积神经网络对图像进行分类存在不足:(1)二维网络结构不能利用图像的三维空间信息,会损失很多有用信息,从而限制了模型的分类性能;(2)卷积神经网络为多层学习网络,传统的方法只是对网络的最后一层进行监督,忽略了中间层监督对模型分类效果的影响。
技术实现思路
本专利技术所要解决的技术问题是,提供一种可以对三维视网膜OCT图像进行分类,提高分类准确性的基于三维卷积神经网络的视网膜OCT图像的分类算法。为解决上述技术问题,本专利技术采用的技术方案为:基于三维卷积神经网络的视网膜OCT图像的分类算法,包括以下步骤:S01:采集三种视网膜OCT图像,分别是以黄斑为中心的视 ...
【技术保护点】
基于三维卷积神经网络的视网膜OCT图像的分类算法,其特征是,包括以下步骤:S01:采集三种三维视网膜OCT图像,分别是以黄斑为中心的视网膜OCT图像、以视神经乳头为中心的视网膜OCT图像和以大视野为中心的视网膜OCT图像,将所述三种视网膜OCT图像每种分类标记为2类,分别是正常视网膜图像和异常视网膜图像;S02:数据预处理,对三维视网膜OCT图像数据进行降采样,得到统一大小的三维图像用于输入三维卷积神经网络;S03:根据迁移学习理论,用大量标记好的自然图像预训练一个三维卷积神经网络模型,所述三维卷积神经网络模型包括输入层、若干卷积层、若干池化层、若干全连接层以及输出层;S04:用预处理好的三维视网膜OCT图像对训练好的三维卷积神经网络模型进行微调,在三维卷积神经网络模型主流网络中间的卷积层后加入分支网络,所述分支网络利用多层感知卷积进一步提取图像局部信息,所述分支网络同样由卷积层、池化层、全连接层和输出层组成,将所述主流网络和所述分支网络的输出层进行融合;S05:将测试图像按照S02步骤进行预处理,利用S04步骤中微调后的所述三维卷积神经网络模型进行测试,输出分类结果。
【技术特征摘要】
1.基于三维卷积神经网络的视网膜OCT图像的分类算法,其特征是,包括以下步骤:S01:采集三种三维视网膜OCT图像,分别是以黄斑为中心的视网膜OCT图像、以视神经乳头为中心的视网膜OCT图像和以大视野为中心的视网膜OCT图像,将所述三种视网膜OCT图像每种分类标记为2类,分别是正常视网膜图像和异常视网膜图像;S02:数据预处理,对三维视网膜OCT图像数据进行降采样,得到统一大小的三维图像用于输入三维卷积神经网络;S03:根据迁移学习理论,用大量标记好的自然图像预训练一个三维卷积神经网络模型,所述三维卷积神经网络模型包括输入层、若干卷积层、若干池化层、若干全连接层以及输出层;S04:用预处理好的三维视网膜OCT图像对训练好的三维卷积神经网络模型进行微调,在三维卷积神经网络模型主流网络中间的卷积层后加入分支网络,所述分支网络利用多层感知卷积进一步提取图像局部信息,所述分支网络同样由卷积层、池化层、全连接层和输出层组成,将所述主流网络和所述分支网络的输出层进行融合;S05:将测试图像按照S02步骤进行预处理,利用S04步骤中微调后的所述三维卷积神经网络模型进行测试,输出分类结果。2.根据权利要求1所述的基于三维卷积神经网络的视网膜OCT图像的分...
【专利技术属性】
技术研发人员:陈新建,刘云,
申请(专利权)人:苏州比格威医疗科技有限公司,
类型:发明
国别省市:江苏,32
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。