[发明专利]一种基于双重边缘表示的眼底OCT图像视网膜分层方法

权利要求书

1.一种基于双重边缘表示的眼底OCT图像视网膜分层方法，其特征在于，包括以下步骤：

步骤1：数据集获取及预处理；

步骤2：构建基于双重边缘表示的眼底OCT图像视网膜分层模型MTDBA-Net；

步骤3：使用灰度归一化训练图像和人工标注结果及双重边缘表示训练提出的MTDBA-Net；

步骤4：通过训练好的模型实现眼底OCT图像视网膜分层。

2.根据权利要求1所述的一种基于双重边缘表示的眼底OCT图像视网膜分层方法，其特征在于，步骤1具体方法如下：

获取训练用眼科OCT图像，并对图像进行人工视网膜层标注以及灰度归一化处理，归一化函数如下：

其中x是原始强度值，是归一化以后的值。

3.根据权利要求2所述的一种基于双重边缘表示的眼底OCT图像视网膜分层方法，其特征在于，步骤2具体方法如下：

所述基于双重边缘表示的眼底OCT图像视网膜分层模型MTDBA-Net为多任务三分支结构，主要由4个部分构成：共享特征提取器、执行分割任务的主分支、执行上边缘表示的上分支以及执行下边缘表示回归任务的下分支；

共享特征提取器采用基于U-net的网络架构，其编码部分采用vgg16,解码部分只保留了前2次上采样所对应的网络架构；三分支结构都共享上述的共享特征提取器，将输入图像经共享特征提取器后的输出作为三分支结构的输入；

所述的主分支包含解码器和分割预测结构两个部分；其中解码器包含三层，每层由3x3卷积，BatchNorm,ReLu构成；分割预测结构由1x1卷积层和softmax函数构成；

所述的上分支由3个部分构成：解码器，边缘预测结构以及上边缘表征模块；边缘预测结构由1x1卷积层和sigmoid函数构成的边缘预测结构以及上边缘表征模块；

所述的下分支由3个部分构成：解码器、边缘预测结构和下边缘表征模块；其中主、上、下三个分支中的解码器结构相同但参数不共享；上、下分支中的边缘预测结构相同但参数并不共享；

上、下边缘表征模块的上、下边缘表征获取，根据如下公式计算：

其中：

其中，和分别表示视网膜内像素点到所属视网膜层上边缘/下边缘的最小欧氏距离；GT^c表示视网膜分层标注属于第c层的像素集合；和分别表示GT^c中像素的上边缘和下边缘；||x-y||表示像素点x和y之间的欧氏距离，min表示最小距离。