超声图像中眼球区域分割方法与流程

本发明涉及人工智能领域，尤其涉及一种超声图像中眼球区域分割方法。

背景技术：

眼科超声图像可以通过常规超声设备采集，为了在眼科超声图像中分割出眼球区域，需要重建眼球的三维结构，才能全面精确地反映后巩膜的具体形态以及与视神经的位置关系，同时，目前大多采用人工方式进行眼球区域的分割操作，不仅效率低，且分割准确度还取决于操作人员的作业水平。

鉴于此，有必要研发一种具有较高准确度的眼球区域自动分割的方案。

技术实现要素：

本发明的目的是提供一种超声图像中眼球区域分割方法，可以提高眼球区域分割效率，并确保分割准确度。

本发明的目的是通过以下技术方案实现的：

一种超声图像中眼球区域分割方法，包括：

利用一定数量的、且经专家进行眼球区域标注的眼科超声图像构建训练集；

分别计算训练集中的每一眼科超声图像的有向符号距离场矩阵，由训练集中的所有图像构成训练集有向符号距离场矩阵；

利用训练集以及有向符号距离场矩阵训练预先搭建的卷积神经网络ss-unet；

将待分割的眼科超声图像输入至训练完毕的卷积神经网络ss-unet，从而完成眼球区域分割。

由上述本发明提供的技术方案可以看出，该方法完全基于2d卷积神经网络，极大的提高了眼球区域分割速度，同时，在数据集上测试结果显示其具有较高的分割准确率。

附图说明

为了更清楚地说明本发明实施例的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本发明的一些实施例，对于本领域的普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他附图。

图1为本发明实施例提供的一种超声图像中眼球区域分割方法的示意图。

具体实施方式

下面结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明的保护范围。

本发明实施例提供一种超声图像中眼球区域分割方法，其主要包括：

1、利用一定数量的、且经专家进行眼球区域标注的眼科超声图像构建训练集。

2、分别计算训练集中的每一眼科超声图像的有向符号距离场矩阵，由训练集中的所有图像构成训练集有向符号距离场矩阵。

3、利用训练集以及有向符号距离场矩阵训练预先搭建的卷积神经网络ss-unet。

4、将待分割的眼科超声图像输入至训练完毕的卷积神经网络ss-unet，从而完成眼球区域分割。

为了便于理解，下面针对上述方案做详细的介绍。

如图1所示，为超声图像中眼球区域分割方法的流程图。

一、构建数据集。

本发明实施例中，可以采集三维眼科超声影像，共多段(例如6段)三维(二维加上时间)影像，对每段三维影像逐帧抽取二维图像，得到多幅(例如，568幅)二维超声图像。

然后，由专家对眼科超声图像中眼球所在区域进行标注，眼球区域标注为黑色，背景区域标注为白色，以便机器学习如何自动进行区域分割。

之后，对于标注过的眼科超声图像进行数据扩充，从而构建数据集；所述数据扩充方式包括：缩放、旋转与平移。例如，对标注过的568幅图中的400幅图进行数据扩充，用缩放，旋转，平移的方法得到额外的数据，以便达到更好的训练效果。这400幅图像经过数据扩充得到4290幅图像作为训练集，而未进行数据扩充的168幅图像则用于测试。

同时，将每一眼科超声图像调整到m×n的大小；假设原始的眼科超声图像为480×470可以将它们统一调节到到96×96大小。

二、计算训练集有向符号距离场矩阵。

根据标注的眼球区域的轮廓曲线，计算每一眼科超声图像的有向符号距离场(sdf)矩阵

其中，ωinside表示标注的眼球区域内部，ωoutside表示标注的背景区域，pi,j表示眼科超声图像上坐标(i,j)处的点，c表示眼球区域的轮廓曲线，d(pi,j,c)表示计算眼科超声图像上的点pi,j到轮廓曲线c的最短欧式距离：

其中，ck表示轮廓曲线c上的第k个点；

对每一眼科超声图像都计算出大小为m×n的有向符号距离场矩阵并展开成m×n×1的有向符号距离场向量，将每一眼科超声图像对应的有向符号距离场向量横向堆叠，最终由训练集中的m张眼科超声图像得到m×n×m大小的训练集有向符号距离场矩阵。

三、搭建卷积神经网络ss-unet

还参见图1，所述卷积神经网络ss-unet包括：特征提取路径、语义嵌入路径和上采样路径。

1、特征提取路径。

特征提取路径包括五个网络阶段，记为downstage1～downstage5；每个网络阶段的输入数据先经过两个卷积模块，每个卷积模块包含1个卷积层，1个归一化层和1个使用relu函数的激活层，随后通过1个池化层后输出，并作为下一个网络阶段的输入；示例性的，downstage1～downstage5输出的特征图数量依次为：32，64，12，256，512。

2、语义嵌入路径。

语义嵌入路径包括两个语义嵌入模块；第一个语义嵌入模块将特征提取路径中网络阶段downstage3、downstage4以及downstage5的输出分别通过1个卷积层和1个采用双线性插值的上采样层后，与网络阶段downstage2的输出逐像素相乘作为网络阶段downstage2的新的输出；第二个语义嵌入模块将特征提取路径中网络阶段downstage4与downstage5的输出分别通过1个卷积层和1个采用双线性插值的上采样层后，与网络阶段downstage3的输出逐像素相乘作为网络阶段downstage3的新的输出。

3、上采样路径。

上采样路径包括五个阶段，记为upstage1～upstage5；上采样路径的阶段upstage5即特征提取路径的网络阶段downstage5；第i个阶段upstagei的输出经过上采样后与网络阶段downstagei-1的输出拼接，再通过2个卷积层作为upstagei-1的输出；示例性的，upstage1～upstage5输出的特征图数量依次为：512，256，128，64，32。

在阶段upstage1输出特征图后，再接一个卷积层将通道数降为1，得到m×n大小的预测输出。

示例性的，卷积神经网络ss-unet中所用的所有卷积核大小都为3×3，卷积步长为1×1。

四、训练卷积神经网络ss-unet。

训练时，将训练集有向符号距离场矩阵中对应于输入眼科超声图像的有向符号距离场向量变形为m×n大小的有向符号距离场矩阵以卷积神经网络ss-unet的预测输出与有向符号距离场矩阵间的均方误差作为网络损失函数：

其中，m、n分别为眼科超声图像的长和宽，xij为卷积神经网络ss-unet的预测输出中第i行第j列处的像素值，yij为有向符号距离场矩阵中第i行第j列处的数值；

求出损失函数值(误差)后，依据链式法则求出其对应于每个神经元的梯度，通过误差反向传播更新参数，学习率由adam算法自动调节；卷积神经网络ss-unet的参数学习过程一直持续到损失函数值收敛为止。

五、测试阶段。

将待分割的眼科超声图像输入至训练完毕的卷积神经网络ss-unet，训练完毕的卷积神经网络ss-unet将输出相应的预测图；对预测图进行二值化，并二值图中所有值大于0的像素重新赋值为0，所有值小于0的像素重新赋值为255，得到对眼球区域的最终分割结果；其中，像素值为0表示属于眼球区域，像素值为255表示属于背景部分。

本发明实施例上述方案，可以应用于医院的影像科，对采集到的眼科超声图像自动进行眼球区域的分割检测，还可以应用于相关的科研机构；本发明实施例上述方案主要是针对眼球区域分割方案进行改进，以提高分割准确度与分割效率，其具体应用方向可以由用户自行选择。在实施上，可以以软件的方式安装于工作电脑，提供实时检测；也可以安装于后台服务器，提供大批量后台检测。

与现有方法相比，本发明实施例提供的上述方案完全基于2d卷积神经网络，在数据集上测试，眼球区域的分割准确度达到：平均像素准确率为98.81％，区域轮廓的平均误差为0.89个像素，匹配失败(区域轮廓误差大于2个像素)的概率为5.95％。

通过以上的实施方式的描述，本领域的技术人员可以清楚地了解到上述实施例可以通过软件实现，也可以借助软件加必要的通用硬件平台的方式来实现。基于这样的理解，上述实施例的技术方案可以以软件产品的形式体现出来，该软件产品可以存储在一个非易失性存储介质(可以是cd-rom，u盘，移动硬盘等)中，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行本发明各个实施例所述的方法。

以上所述，仅为本发明较佳的具体实施方式，但本发明的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本发明披露的技术范围内，可轻易想到的变化或替换，都应涵盖在本发明的保护范围之内。因此，本发明的保护范围应该以权利要求书的保护范围为准。