本发明属于图像处理技术领域,更为具体地讲,涉及一种基于插值卷积神经网络的低剂量ct重建方法。
背景技术:
目前,人们对ct辐射剂量对人体可能造成的潜在危害问题越来越重视。通过降低管电流强度(低剂量ct)和减少采样的次数(稀疏角度ct),能够实现降低辐射剂量。但这样破坏了投影数据的完备性,由传统重建算法直接重建出的图像质量会严重退化。因此,如何在降低辐射剂量的同时保证重建图像的质量成为近年来ct研究的一个热点。
针对于低剂量投影数据中的噪声的问题和重建图像中出现条形伪影和噪声的问题,从解决方法上,可以分为投影域数据恢复和图像后处理方法。本发明针对投影域的噪声,提出一种基于插值卷积神经网络的低剂量ct重建方法。降低采样角度后,对角度缺失的投影数据进行插值后再进行滤波反投影重建,得到重建的ct图像。
随着深度学习的发展,将卷积神经网络应用于投影数据的恢复引起了广大研究者的兴趣。正弦图像中的每一列都对应于一个角度的投影值,降低采样角度后,正弦图信息缺失,重建的ct图像会出现伪影。本说明受到基于卷积神经网络的超分辨率技术的启发,将超分辨率技术应用于投影数据重建,重建完备的投影数据,优化ct重建效果。
超分辨率就是有一张尺寸比较小的图,当我们将其放大的时候,会出现模糊的现象,超分辨率重建就是将放大之后的图像尽可能地清晰。espcn又称亚像素卷积神经网络,是一种应用于超分辨率分析的神经网络模型,可以直接在低分辨率图像尺寸上提取特征,计算得到高分辨率图像。利用超分辨率网络在性能上的优势,对角度缺失的投影数据进行插值后再进行滤波反投影重建,得到重建的ct图像。有了更加完整的投影数据,可以得到更好的重建效果。
技术实现要素:
本发明的目的在于克服现有技术的不足,提供一种基于插值卷积神经网络的低剂量ct重建方法,利用超分辨率网络在性能上的优势,对角度缺失的投影数据进行插值后再进行滤波反投影重建,得到重建的ct图像。
为实现上述发明目的,本发明一种基于插值卷积神经网络的低剂量ct重建方法,其特征在于,包括以下步骤:
(1)、构建训练数据集
(1.1)、采集不同患者的标准剂量ct图像,对每张图像进行radon变换,得到标准剂量的投影正弦图;
(1.2)、将投影正弦图量化至(0,1),然后提取其奇数列,得到模拟的低剂量投影正弦图,剩余的偶数列作为期望插值图;
(1.3)、将低剂量投影正弦图与期望插值图用大小为m×n的滑窗,按纵向步长m-h,横向步长n-h进行滑动分块,滑动分块后得到若干组m×n的低剂量投影正弦图块与期望插值图像块,其中h为偶数,h=l1 l2 l3-3,l1,l2,l3为插值卷积神经网络模型中各层卷积核的尺寸;
(1.4)、对每一组m×n的期望插值图像块进行边界裁剪,截取中心为(m-h)×(n-h)的区域,得到样本期望插值图像块;
(1.5)、将每一组低剂量投影正弦图记为xi,每一组样本期望插值图块记为yi,然后将其构成训练数据集中的一个样本记为{xi,yi},下标i表示样本组数;
(2)、构建插值卷积神经网络模型
插值卷积神经网络模型由三层卷积层组成,其中,第一层采用c1个l1×l1卷积核,滤波步长为1,第二层采用c2个l2×l2×c1的卷积核,步长为1,第三层采用1个l3×l3×c2的卷积核,步长为1,每层卷积后取有效区并采用sigmoid函数激活;
(3)、训练插值卷积神经网络模型
训练数据集中选取一组训练样本{xi,yi},将xi输入至插值卷积神经网络模型,经第一层卷积后得到c1张(m-l1 1)×(n-l1 1)的特征图,再经过第二层卷积后得到c2张(m-l1-l2 2)×(n-l1-l2 2)的特征图,最后经过第三层卷积后输出一张(m-h)×(n-h)的预测插值图yi*;然后计算预测插值图yi*与样本期望插值图yi之间的像素误差的平方和,以像素误差的平方和为损失值,利用梯度下降算法对插值卷积神经网络进行权重更新;再选取下一组训练样本重复上述步骤,直至遍历所有训练样本完成一轮训练;判断所有训练样本的累积像素误差平方和是否小于预设阈值,如果小于,则训练结束,得到训练完成的插值卷积神经网络模型;否则,进行下一轮训练,直到累积像素误差的平方和小于预设阈值时,训练结束;
(4)、利用插值卷积神经网络重建低剂量ct图像
(4.1)、实时ct扫描时,仅在标准剂量ct的奇数投影角度进行采样,获得低剂量投影正弦图po,设其大小为m1×n1;
(4.2)、对低剂量投影正弦图po进行列扩展:复制原图po的前1~(h/2)列到po的末端,扩展出第(n1 1)~(n1 h/2)列,再复制原图po最后的(n1-h/2 1)~n1列到po的前端,扩展出第1~(h/2)列,从而得到列扩展后的低剂量投影正弦图p1;
(4.3)、对低剂量投影正弦图p1进行行扩展,其中上下各扩展(h/2)行,扩展行的元素值均用零填充,最终得到(m1 h)×(n1 h)的扩展正弦图pg;
(4.4)、将扩展正弦图pg输入至训练好的插值卷积神经网络,得到偶数投影角度的预测正弦图pe,其大小为m1×n1;
(4.5)、以低剂量投影正弦图po各列为奇数行,预测正弦图pe各列依次作为偶数行,组合得到m1×(2n1)的标准剂量投影正弦图p;
(4.6)、将标准剂量投影正弦图p量化至原图像空间,并进行反radon变换,得到重建的ct图像。
本发明的发明目的是这样实现的:
本发明为一种基于插值卷积神经网络的低剂量ct重建方法,利用插值卷积神经网络在超分辨率重建上的应用,通过插值卷积神经网络预测ct投影正弦图上缺失的投影数据,再加上已知的投影数据,得到完备的ct投影正弦图,最后再经过滤波反投影得到重建的ct图像,这样可以有效地去除低剂量ct图像上的噪声,得到更好的重建效果。
同时,本发明基于插值卷积神经网络的低剂量ct重建方法还具有以下有益效果:
(1)、本发明受到插值卷积神经网络在超分辨率应用上的启发,在ct图像的正弦域上重建完备的投影数据,再经过滤波反投影算法得到重建的ct图像,可以有效地去除低剂量ct图像上的噪声,得到更好的重建效果;
(2)、本发明仅需要用神经网络预测缺失部分的投影数据,然后与已知的投影数据组合,可以得到完备的投影数据图像;相较于用神经网络直接输出完备的投影数据图像,可以减小网络计算量,加快了速度;
(3)、本发明采用全卷积网络作为去噪网络,这样不受输入图像尺寸的影响,可以直接输入完整的ct图像进行去噪,这样使用图像块进行训练可以加快网络的训练速度;
(4)、完整的ct图像数量有限,可能导致训练的网络过拟合,但将完整的ct图像裁剪为图像块,增加训练样本的数量,可以产生更好的学习效果,增加网络模型稳定性。
附图说明
图1是本发明一种基于插值卷积神经网络的低剂量ct重建方法流程图;
图2是构建训练数据集的流程图;
图3是利用插值卷积神经网络重建低剂量ct图像的流程图。
具体实施方式
下面结合附图对本发明的具体实施方式进行描述,以便本领域的技术人员更好地理解本发明。需要特别提醒注意的是,在以下的描述中,当已知功能和设计的详细描述也许会淡化本发明的主要内容时,这些描述在这里将被忽略。
实施例
图1是本发明一种基于插值卷积神经网络的低剂量ct重建方法流程图。
在本实施例中,如图1所示,本发明一种基于插值卷积神经网络的低剂量ct重建方法,包括以下步骤:
s1、构建训练数据集,具体如图2所示,包括以下步骤:
s1.1、采集不同患者的标准剂量ct图像,对每张图像进行radon变换,得到标准剂量的投影正弦图;
s1.2、将投影正弦图量化至(0,1),从而得到512×512的标准剂量投影正弦图;
s1.3、提取其奇数列,得到模拟的低剂量投影正弦图,大小为512×256,剩余的偶数列作为期望插值图,大小为512×256;
s1.4、将低剂量投影正弦图与期望插值图用大小为m×n=16×16的滑窗,按纵向步长m-h=8,横向步长n-h=8进行滑动分块,滑动分块后得到1953组16×16的低剂量投影正弦图块与期望插值图像块;
s1.5、对每一组16×16的期望插值图像块进行边界裁剪,截取中心为8×8的区域,得到样本期望插值图像块;
s1.6、将每一组低剂量投影正弦图记为xi,每一组样本期望插值图块记为yi,然后将其构成训练数据集中的一个样本记为{xi,yi},下标i表示样本组数,最终每张ct切片形成1953个训练样本;
s2、构建插值卷积神经网络模型
插值卷积神经网络模型由三层卷积层组成,其中,第一层采用c1=64个l1×l1=5×5卷积核,滤波步长为1,第二层采用c2=32个l2×l2×c1=3×3×64的卷积核,步长为1,第三层采用1个l3×l3×c2=3×3×32的卷积核,步长为1,每层卷积后取有效区并采用sigmoid函数激活;
s3、训练插值卷积神经网络模型
训练数据集中选取一组训练样本{xi,yi},将xi输入至插值卷积神经网络模型,对于16×16的输入图像,经第一层卷积后得到64张12×12的特征图;经第二层卷积后得到32张10×10的特征图;经第三层卷积后最终输出1张8×8的预测插值图yi*;然后计算预测插值图yi*与样本期望插值图yi之间的像素误差的平方和,以像素误差的平方和为损失值,利用梯度下降算法对插值卷积神经网络进行权重更新;再选取下一组训练样本重复上述步骤,直至遍历所有训练样本完成一轮训练;判断所有训练样本的累积像素误差平方和是否小于预设阈值,如果小于,则训练结束,得到训练完成的插值卷积神经网络模型;否则,进行下一轮训练,直到累积像素误差的平方和小于预设阈值时,训练结束;
s4、利用插值卷积神经网络重建低剂量ct图像,具体如图3所示,包括以下步骤:
s4.1、实时ct扫描时,仅在标准剂量ct的奇数投影角度进行采样,获得低剂量投影正弦图po,设其大小为m1×n1=512×256;
s4.2、对低剂量投影正弦图po进行列扩展:复制原图po的前1~4列到po的末端,扩展出第257~260列,再复制原图po最后的253~256列到po的前端,扩展出第1~4列,从而得到列扩展后的低剂量投影正弦图p1;
s4.3、对低剂量投影正弦图p1进行行扩展,其中上下各扩展4行,扩展行的元素值均用零填充,最终得到520×264的扩展正弦图pg;
s4.4、将扩展正弦图pg输入至训练好的插值卷积神经网络,得到偶数投影角度的预测正弦图pe,其大小为512×256;
s4.5、以低剂量投影正弦图po各列为奇数行,预测正弦图pe各列依次作为偶数行,组合得到512×512的标准剂量投影正弦图p;
s4.6、将标准剂量投影正弦图p量化至原图像空间,并进行反radon变换,得到重建的ct图像。
尽管上面对本发明说明性的具体实施方式进行了描述,以便于本技术领域的技术人员理解本发明,但应该清楚,本发明不限于具体实施方式的范围,对本技术领域的普通技术人员来讲,只要各种变化在所附的权利要求限定和确定的本发明的精神和范围内,这些变化是显而易见的,一切利用本发明构思的发明创造均在保护之列。
1.一种基于插值卷积神经网络的低剂量ct重建方法,其特征在于,包括以下步骤:
(1)、构建训练数据集
(1.1)、采集不同患者的标准剂量ct图像,对每张图像进行radon变换,得到标准剂量的投影正弦图;
(1.2)、将投影正弦图量化至(0,1),然后提取其奇数列,得到模拟的低剂量投影正弦图,剩余的偶数列作为期望插值图;
(1.3)、将将低剂量投影正弦图与期望插值图用大小为m×n的滑窗,按纵向步长m-h,横向步长n-h进行滑动分块,滑动分块后得到若干组m×n的低剂量投影正弦图块与期望插值图像块,其中h为偶数,h=l1 l2 l3-3,l1,l2,l3为插值卷积神经网络模型中各层卷积核的尺寸;
(1.4)、对每一组m×n的期望插值图像块进行边界裁剪,截取中心为(m-h)×(n-h)的区域,得到样本期望插值图像块;
(1.5)、将每一组低剂量投影正弦图记为xi,每一组样本期望插值图块记为yi,然后将其构成训练数据集中的一个样本记为{xi,yi},下标i表示样本组数;
(2)、构建插值卷积神经网络模型
插值卷积神经网络模型由三层卷积层组成,其中,第一层采用c1个l1×l1卷积核,滤波步长为1,第二层采用c2个l2×l2×c1的卷积核,步长为1,第三层采用1个l3×l3×c2的卷积核,步长为1,每层卷积后取有效区并采用sigmoid函数激活;
(3)、训练插值卷积神经网络模型
从训练数据集中选取一组训练样本{xi,yi},将xi输入至插值卷积神经网络模型,经第一层卷积后得到c1张(m-l1 1)×(n-l1 1)的特征图,再经过第二层卷积后得到c2张(m-l1-l2 2)×(n-l1-l2 2)的特征图,最后经过第三层卷积后输出一张(m-h)×(n-h)的预测插值图
(4)、利用插值卷积神经网络重建低剂量ct图像
(4.1)、实时ct扫描时,仅在标准剂量ct的奇数投影角度进行采样,获得低剂量投影正弦图po,设其大小为m1×n1;
(4.2)、对低剂量投影正弦图po进行列扩展:复制原图po的前1~(h/2)列到po的末端,扩展出第(n1 1)~(n1 h/2)列,再复制原图po最后的(n1-h/2 1)~n1列到po的前端,扩展出第1~(h/2)列,从而得到列扩展后的低剂量投影正弦图p1;
(4.3)、对低剂量投影正弦图p1进行行扩展,其中上下各扩展(h/2)行,扩展行的元素值均用零填充,最终得到(m1 h)×(n1 h)的扩展正弦图pg;
(4.4)、将扩展正弦图pg输入至训练好的插值卷积神经网络,得到偶数投影角度的预测正弦图pe,其大小为m1×n1;
(4.5)、以低剂量投影正弦图po各列为奇数行,预测正弦图pe各列依次作为偶数行,可以组合得到m1×(2n1)的标准剂量投影正弦图p;
(4.6)、将标准剂量投影正弦图p量化至原图像空间,并进行反radon变换,得到重建的ct图像。
技术总结