《眼科新进展》
0 引言
1 数据和方法
1.1 数据
1.2 方法
1.2.1 构建样本数据集
1.2.2 搭建的图像分类网络的结构
(1) 输入层。
(2) 卷积层和激活函数。
(3) 池化层。
(4) Dropout层。
(5) 全连接层和输出层。
1.2.3 训练分类网络
2 结果
2.1 模型的评估指标
2.2 实验结果
3 讨论
4 结论
文章摘要:目的提出一种基于卷积神经网络(convolutional neural network, CNN)的眼科光学相干断层成像(optical coherence tomography, OCT)图像自动分类方法,实现对视网膜OCT图像的自动分类,缓解人工诊断依赖医生的临床经验、费时费力等问题。方法基于公开的数据集2014BOESrinivasan构建了2个样本数据集。其中样本数据集一为仅对数据集中的图像进行预处理后裁剪,样本数据集二为对取出测试集后剩余图像的裁剪过程中引入随机平移和水平翻转技术对图像进行扩充,并划分为训练集和验证集。搭建基于CNN的视网膜OCT图像分类网络,并分别使用两个数据集训练网络得到分类模型。最后使用独立的测试集对模型进行测试,并通过输出混淆矩阵查看模型对3种类别图像的分类情况。结果通过混淆矩阵计算得出,使用扩充后的图像训练的分类模型的准确度为93.43%,灵敏度为91.38%,特异度为95.88%。结论提出的基于CNN的视网膜OCT图像自动分类方法可以对老年性黄斑变性、糖尿病性黄斑水肿和正常3种类别的视网膜OCT图像进行分类。同时,数据扩充有助于提高分类算法的性能。
文章关键词: