[发明专利]基于3D全卷积神经网络的脊柱分割方法

摘要

本发明公开了一种基于3D全卷积神经网络的脊柱分割方法。本发明包括如下步骤：步骤1.建一个用于脊柱分割的3D全卷积神经网络；步骤2.准备数据集，数据集包括训练集+测试集；步骤3.利用训练集训练神经网络，得到网络模型；步骤4.使用得到的网络模型分割测试集中脊柱CT数据，得到分割结果。本发明借助于3D全卷积神经网络来分割脊柱，不仅是全自动的，而且分割精度很准确。

主权项

1.基于3D全卷积神经网络的脊柱分割方法，其特征在于包括如下步骤：步骤1.建一个用于脊柱分割的3D全卷积神经网络；步骤2.准备数据集，数据集包括训练集+测试集；步骤3.利用训练集训练神经网络，得到网络模型；步骤4.使用得到的网络模型分割测试集中脊柱CT数据，得到分割结果；步骤1所述的3D全卷积神经网络的构建，具体如下：卷积神经网络的输入是一个大小为128×128×64的体块，输出也是128×128×64的体块；网络结构分为左右两个部分，网络的左侧是一个压缩的过程，右侧是一个扩张的过程；除了降采样和上采样之外，卷积层均采用5×5×5的卷积核进行卷积，步长为1；左侧的压缩的过程被分为5个不同的阶段，每个阶段的特征通道的尺寸不同，通道数也不同，都包括两个卷积层；每个卷积层都采用PRelu非线性激活函数；每个阶段的最后通过执行卷积操作来降低分辨率，卷积时采用2×2×2的卷积核，步长为2，进行降采样；随着压缩阶段特征通道的尺寸减小为原来的1/2，提取的特征的通道数为原来的2倍；具体来说：左侧第一个阶段的通道数为16，每个特征通道的尺寸为128×128×64；第二个阶段通道数为32，每个特征通道的尺寸是64×64×32，第三个阶段的通道数为64，每个特征通道的尺寸是32×32×16，第四个阶段的通道数为128，每个特征通道的尺寸为16×16×8，第五个阶段的通道数为256，每个特征通道的尺寸为8×8×4；网络的右侧的扩张阶段被分为不同的4个阶段，每个阶段的特征通道的尺寸不同，通道数也不同，每个阶段包含两个卷积层；随着特征通道的尺寸变为原来的2倍，通道数变为原来的1/2；具体来说：右侧第一个阶段是通过左侧的第五个阶段，直接通过2×2×2的卷积核降采样得到的，通道数为256，每个特征通道的尺寸是16×16×8，第二个阶段的通道数为128，每个特征通道的尺寸是32×32×16，第三个阶段的通道数为64，特征通道的尺寸为64×64×32，第四个阶段的通道数为32，特征通道的尺寸为128×128×64；除第四个阶段的最后，每个阶段的最后都会采用2×2×2的卷积核进行反卷积操作，使特征通道的尺寸变为原来的2倍；右侧的第四个阶段得到的32通道的结果，通过最后一次的卷积层，卷积核的尺寸为1×1×1，计算出两个128×128×64的特征图，再通过尺寸为1×1×1的卷积核来产生和输入的尺寸相同的输出，转化为前景和背景的概率，通过softmax层来实现；左侧的每个阶段的输入被用作卷积，但同时被添加到本阶段第二个卷积层的输出上，来学习一个残差函数；左侧每个阶段经过两个卷积层卷积后的结果，会和右侧扩张阶段特征通道尺寸相同的阶段的上一个阶段降采样后的结果，同时作为特征通道数据，进行扩张阶段的卷积操作。