一种适用于中子成像系统图像的基于高斯‑泊松混合噪声模型的图像复原方法与流程

本发明属于信息处理技术领域，涉及一种图像降噪方法。

背景技术：

中子成像技术是用于非破坏性测试及其他许多基础研究领域的一种有效工具。在可移动中子成像系统中，由于高达10⁹n/s的中子发生器，小于15m的校直系统，慢中子流和受限的长度/直径比值，所得图像通常伴有严重的高斯泊松混合噪声。图像噪声会造成：①图像主观视觉效果差；②图像特征被掩盖，降低图像质量和精度，影响对图像进行特征提取和分析等后期工作。故对于中子成像系统，图像降噪是一个至关重要的后期处理过程。

图像的信息主要分为高频和低频两个部分，高频部分是图像中像素亮度值变换较快的部分，主要包含图像中的轮廓信息，表现图像的主体特征。低频部分是图像中较为平滑的部分，表现了图像的细节特征。对于噪声而言，也属于图像中高频的部分。由于图像边缘特征反映图像主题内容，直接冲击人的视觉感受，所以图像降噪处理过程中应尽量保持图像的边缘特征。而在传统的bm3d降噪方法中，由于使用全局相似块匹配，相似块匹配效果较差，在较高方差的高斯噪声干扰下，滤波过程常造成图像中物体边缘和噪声一起作为高频信息被削减或去除，造成物体边缘模糊严重，导致最终降噪效果不良。

技术实现要素：

本发明的目的是提出一种适用于中子成像系统图像的基于高斯-泊松混合噪声模型的图像复原方法。

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

本发明所述的一种适用于中子成像系统图像的基于高斯-泊松混合噪声模型的图像复原方法，其特征是包括以下步骤：

对中子成像图像的降噪问题进行数学建模，设中子成像系统直接得到的图像为zi＝pi(yi)+ni，其中ni为加性的方差为σ²的高斯白噪声，pi(yi)为乘性的参数为yi的泊松噪声项。其中泊松噪声项的参数yi即为无噪图像i位置的像素亮度值。中子成像图像的降噪问题就是从zi中尽可能恢复出yi。降噪质量可使用降噪后图像与无噪图像的psnr值进行评估，此值越高则降噪效果越优秀。

s1：对要处理的原始图像进行gat变换；

s2：对变换后的图像使用bm3d算法stage1处理，得到基本估计图像；

s3：对s2所得基本估计图像使用k-均值聚类算法划分区域；

s4：在各区域内使用bm3d算法stage2处理(即将stage2的相似块匹配限制在同一图像区域内)，得到最终估计图像；

s5：对最终估计图像做无偏gat逆变换得到复原后的图像。

本发明步骤s3所述的区域划分，有多种方法，本发明优选以下方法：

一种用于bm3dstage1与stage2阶段之间的，基于物体边缘特征的图像区域划分方法，也即一种k-均值聚类算法。k-均值聚类算法使用图像中的随机像素点作为初始值进行迭代。迭代过程使用lloyd算法进行：

①在图像中选取随机分布的k个像素作为聚类中心点初始值；

②计算每个聚类中心点周围n/k×n/k范围内的像素亮度中位值，以这些中位值作为新的聚类中心点；

③检查迭代过程中心点移动距离是否小于给定阈值，若不小于则回到步骤②，否则认为迭代过程收敛，进入下一步；

④以收敛后得到的中心点对图像进行聚类，对某个图像中的像素，计算其与每个中心点的亮度差异值，将此像素归入与之亮度最接近的中心点。

如此将图像划分为以物体边缘划分的k个区域。通常k值与画面中物体数量成正相关。

本发明主要目的是解决传统bm3d算法在强噪声下边缘信息丢失严重的技术问题，提出一种改进的中子成像图像降噪算法，有效降低图像噪声，提高中子成像系统的图像质量。

更具体地说，本发明通过以下技术方案来实现。

本发明所述的一种适用于中子成像系统图像的基于高斯-泊松混合噪声模型的图像复原方法，利用gat变换将高斯泊松混合噪声稳定化为高斯噪声，并利用区域限制bm3d方法减少变换后图像的噪声和模糊，最后使用gat的无偏逆变换得到最终的图像。具体实施方法包括如下步骤：

s1：对要处理的原始图像z进行gat变换

其中σ²为高斯噪声方差，以将高斯泊松混合噪声变换化为高斯白噪声，便于后续bm3d算法处理。

s2：对变换后的图像使用bm3d算法stage1处理，包含以下步骤：

①对给定的图像块，找出图像中的相似的块，将它们组合进一个相似块分组；

②对所得相似块分组进行三维线性变换，三维线性变换算法为bior3d；

③使用硬阈值滤波对分组进行滤波，滤波公式为

④进行步骤②所做三维线性变换的逆变换；

⑤将每个图块对应的三维图像数据组进行加权平均，并恢复至原图像中对应位置。

如此得到基本估计图像。

s3：对基本估计图像使用k-均值聚类算法(本发明使用lloyd算法进行迭代)划分区域，具体步骤如下：

①在图像中选取随机分布的k个像素作为聚类中心点初始值；

②计算每个聚类中心点周围n/k×n/k范围内的像素亮度中位值，以这些中位值作为新的聚类中心点；

③检查迭代过程中心点移动距离是否小于给定阈值ε，若不小于则回到步骤②，否则认为迭代过程收敛，进入下一步；

④以收敛后得到的中心点对图像进行聚类，对某个图像中的像素，计算其与每个中心点的亮度差异值，将此像素归入与之亮度最接近的中心点。

如此对图像中所有像素进行聚类，即可得到基于物体边缘特征的k个区域划分。

s4：在各个区域内的使用bm3d算法stage2阶段进行处理(基于不规则形状的相似块匹配)。在进行相似块匹配时，对s2所得基本估计图像进行方形图块分块，对于跨越两个或两个以上区域的图块，在进行相似块匹配时，对此图块内的像素与一个相同大小的掩码块取交集后，以图块距离公式

为基础进行相似块匹配。当某个图块跨越两个或两个以上图像区域时，使用的掩码m(si)如图4所示。

stage2在匹配后进行的处理算法与stage1相似，但是在进行3d变换时使用dct-3d变换，并且在滤波时使用维纳滤波公式

得到最终估计图像。

s5：对最终估计图像做无偏gat逆变换得到复原后的图像。

本发明通过在bm3d算法的stage2阶段引入在不规则形状区域内限制匹配的相似块匹配算法，解决了传统bm3d算法中不良匹配的问题，更加有效地在高方差噪声条件下保留了图像中物体边缘特征，提升了最终降噪图像的psnr值。

附图说明

图1为本发明所述的图像降噪方法的流程示意图。

图2为本发明所述降噪方法中区域限制bm3d算法的流程示意图。

图3为本发明所述降噪方法中k-均值聚类算法的流程示意图。

图4为本发明所述降噪方法中区域限制bm3dstage2相似块匹配掩码示意图。

具体实施方式

本发明将结合附图，通过以下实施例作进一步说明。

本发明所述的一种适用于中子成像系统图像的基于高斯-泊松混合噪声模型的图像复原方法，包括以下步骤：

步骤(1)：高斯泊松混合噪声是一种乘性噪声和加性噪声叠加的噪声，对于含有乘性噪声的图像而言，要降低图像噪声通常有两种方式，一种是直接考虑噪声模型并据此使用特定的降噪算法，另一种降噪处理思想是使用方差稳定化将带有乘性噪声的图像转化为加性噪声图像(即vst变换)。对带有高斯泊松混和噪声的图像而言，通常使用的变换算法是gat算法。设中子成像系统直接得到的图像为zi＝pi(yi)+ni，其中ni为加性的方差为σ²的高斯白噪声，pi(yi)为乘性的参数为yi的泊松噪声项，则变换

将带有高斯泊松混合噪声的图像变换为被方差为σ²的高斯白噪声污染的加性噪声图像。

步骤(2)：对上一步所得含噪图像使用bm3d算法stage1计算基本估计图像。

首先进行图像相似块搜索，对给定n/k×n/k像素大小的图像块p，搜索方式为：定义相似块集合p(p)＝{q:d(p,q)≤τ^hard}，其中

τ^hard为给定硬阈值，定义相似块组p(p)为距离图块p最近的n^hard个图块。

对p(p)进行3d变换及硬阈值滤波其中为三维线性变换(由块内2dbior算法及块间1d变换构成)，其中

对滤波后的相似块分组进行加权平均，还原至原图像位置，stage1所得基本估计图像为：

步骤(3)：对基本估计图像使用k-均值聚类方法划分区域。k-均值聚类使用权重lloyd算法：在图像中选取随机分布的k个像素作为聚类中心点初始值，计算每个聚类中心点周围n/k范围内的像素亮度中位值，以这些中位值作为新的聚类中心点。重复进行迭代过程。当迭代过程中心点移动距离小于给定阈值时，则认为迭代过程停止，以此中心点对图像进行聚类，将某个像素归入亮度最接近的中心点。

步骤(4)：在区域内部进行bm3dstage2操作。stage2与stage1基本相同，在stage1所得的基本估计图像基础上，仅在同一区域内部进行相似块，但stage2与stage1不同，stage2对相似块组使用的滤波算法为维纳滤波算法，公式为：

其中

步骤(4)所得维纳估计图像为：

步骤(5)：对bm3d处理后的图像进行gat逆变换得到最后的降噪图像。但若将bm3d算法处理后的图象若使用代数ga逆变换则引入均值偏差。故本专利使用的算法是无偏gat逆变换，其近似表达式为