[发明专利]基于CAS-Net的OCT图像视网膜脱离和劈裂自动分割方法

说明书

本发明公开了一种基于CAS‑Net的OCT图像视网膜脱离和劈裂自动分割方法，包括以下步骤：S1、获取OCT图像，将OCT图像分为训练集、验证集和测试集；S2、构建分类辅助分割网络用于OCT图像的分类；分类辅助分割网络包括分类网络和分割网络，分割网络包括全维度特征融合模块和特征融合解码模块，分类网络用于全局平均池化和卷积，并输出OCT图像中多个类别的存在概率；S3、构建整个分类辅助分割网络的损失函数，利用训练集和验证集对分类辅助分割网络进行训练；S4、利用测试集对训练后的分类辅助分割网络进行测试，实现对OCT图像的分割。本发明基于CAS‑Net的OCT图像视网膜脱离和劈裂自动分割方法能够准确的分割出OCT图像中的RD、ORS、MRS和IRS。

技术领域

本发明涉及OCT图像分割技术领域，特别涉及一种基于CAS-Net的OCT图像视网膜脱离和劈裂自动分割方法。

背景技术

近视可引起许多的视网膜病理变化，如黄斑孔、脉络膜新生血管、视网膜萎缩、视网膜脱离(Retinal Detachment，RD)和视网膜劈裂(Retinoschisis，RS)。RD和RS是高度近视中最常见的并发症。对于高度近视患者，视网膜黄斑区域的RD和RS严重损害了视觉功能。光学相干断层成像(Optical Coherence Tomography,OCT)可以清晰显示RD和RS的截面，并且是非侵入式的。

视网膜劈裂以视网膜神经上皮层分裂为特征。根据它发生在视网膜层的位置，视网膜劈裂可分为外层劈裂(Outer Retinoschisis，ORS)、中层劈裂(MiddleRetinoschisis，MRS)和内层劈裂(Inner Retinoschisis，IRS)。视网膜脱离是指视网膜神经上皮层和视网膜色素上皮的分离。RS通常发生在早期阶段。随着疾病的发展，将发生更多的RS并且区域将变大。再更进一步，将会发生RD，一旦发生RD，就需要进行手术治疗。RD、ORS、MRS和IRS的定量对于病理性近视的诊断、分期、管理和术后评估具有重要意义。然而，OCT图像中RD、ORS、MRS和IRS的分割面临着一些挑战：(1)类别分布不均匀，因为在一张OCT图像，不是所有类别都一定出现，(2)目标区域变化较大，一些很小，一些在OCT图像中具有很大的跨度，并且(3)不同种类目标的形状和亮度具有相似性。到目前为止还没有人实现OCT图像中视网膜脱离、外层劈裂、中层劈裂和内层劈裂的全自动分割。

随着卷积神经网络(Convolutional Neural Network，CNN)的兴起，它已被越来越多的用于OCT图像中病灶区域分割任务。其出色的特征提取能力带来了优越的分割性能。捕获上下文信息对于图像分割任务至关重要。除了局部上下文之外，表示全局上下文的长距离依赖关系在处理大的目标区域和在多类分类过程中识别不同的目标类别时尤为重要。一些网络使用全局池化、金字塔式池化或空洞卷积操作来获取有用的上下文信息，但是这只能获得某个像素周围的上下文信息，提高网络的局部感知。还有一些网络通过建立一个点跟所有点之间的联系来获取全局的上下文的长期依赖关系，但这是以巨大的计算复杂度和内存消耗为代价。大多数分割网络对于多类分割任务结果不佳，因为其对于图像中不同类别的鉴别能力比较弱。

发明内容

本发明要解决的技术问题是提供一种可对OCT图像中RD、ORS、MRS和IRS病灶区域进行自动分割、且分割精确度高的基于CAS-Net的OCT图像视网膜脱离和劈裂自动分割方法。

为了解决上述问题，本发明提供了一种基于CAS-Net的OCT图像视网膜脱离和劈裂自动分割方法，其包括以下步骤：

S1、获取OCT图像，将所述OCT图像分为训练集、验证集和测试集；

S2、构建分类辅助分割网络用于OCT图像的分类；所述分类辅助分割网络包括分类网络和分割网络，所述分割网络包括全维度特征融合模块和特征融合解码模块，通过所述全维度特征融合模块融合通道和空间维度的上下文信息，通过所述特征融合解码模块融合所述分类网络和分割网络的特征，所述分类网络用于全局平均池化和卷积，并输出OCT图像中多个类别的存在概率；