一种深度图像空洞的自动填充方法与流程

文档序号:11144888阅读:来源:国知局

技术特征:

1.一种深度图像空洞的自动填充方法,其特征在于,该方法包括以下具体步骤:

步骤1:将空洞分为平滑区空洞和非平滑区空洞

计算图像中任意像素与其周围邻域像素的结构相似性,再根据多尺度结构相似性,将图像中空洞分为平滑区域空洞和非平滑区域空洞;具体地:

利用公式(1)计算图像中任意空洞像素p与它的任意3×3邻域中的像素qi(i=1,2...8)的结构相似性SSIM(p,qi);其中p的3×3邻域记为B1,qi的3×3邻域记为Bi

SSIM(p,qi)=[l(B1,Bi)]α·[c(B1,Bi)]β·[s(B1,Bi)]γ (1)

其中α,β,γ是预设的权重,设为α=1,β=1,γ=1;l(B1,Bi)是亮度比较函数,计算为:

<mrow> <mi>l</mi> <mrow> <mo>(</mo> <msub> <mi>B</mi> <mn>1</mn> </msub> <mo>,</mo> <msub> <mi>B</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>=</mo> <mfrac> <mrow> <mn>2</mn> <msub> <mi>u</mi> <msub> <mi>B</mi> <mn>1</mn> </msub> </msub> <msub> <mi>u</mi> <msub> <mi>B</mi> <mi>i</mi> </msub> </msub> <mo>+</mo> <msub> <mi>C</mi> <mn>1</mn> </msub> </mrow> <mrow> <msubsup> <mi>u</mi> <msub> <mi>B</mi> <mn>1</mn> </msub> <mn>2</mn> </msubsup> <mo>+</mo> <msubsup> <mi>u</mi> <msub> <mi>B</mi> <mi>i</mi> </msub> <mn>2</mn> </msubsup> <mo>+</mo> <msub> <mi>C</mi> <mn>1</mn> </msub> </mrow> </mfrac> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>2</mn> <mo>)</mo> </mrow> </mrow>

c(B1,Bi)是对比度比较函数,计算为:

<mrow> <mi>c</mi> <mrow> <mo>(</mo> <msub> <mi>B</mi> <mn>1</mn> </msub> <mo>,</mo> <msub> <mi>B</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>=</mo> <mfrac> <mrow> <mn>2</mn> <msub> <mi>&sigma;</mi> <msub> <mi>B</mi> <mn>1</mn> </msub> </msub> <msub> <mi>&sigma;</mi> <msub> <mi>B</mi> <mi>i</mi> </msub> </msub> <mo>+</mo> <msub> <mi>C</mi> <mn>2</mn> </msub> </mrow> <mrow> <msubsup> <mi>&sigma;</mi> <msub> <mi>B</mi> <mn>1</mn> </msub> <mn>2</mn> </msubsup> <mo>+</mo> <msubsup> <mi>&sigma;</mi> <msub> <mi>B</mi> <mi>i</mi> </msub> <mn>2</mn> </msubsup> <mo>+</mo> <msub> <mi>C</mi> <mn>2</mn> </msub> </mrow> </mfrac> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>3</mn> <mo>)</mo> </mrow> </mrow>

s(B1,Bi)是结构比较函数,计算为:

<mrow> <mi>s</mi> <mrow> <mo>(</mo> <msub> <mi>B</mi> <mn>1</mn> </msub> <mo>,</mo> <msub> <mi>B</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>=</mo> <mfrac> <mrow> <mi>&sigma;</mi> <mo>+</mo> <msub> <mi>C</mi> <mn>3</mn> </msub> </mrow> <mrow> <msub> <mi>&sigma;</mi> <msub> <mi>B</mi> <mn>1</mn> </msub> </msub> <msub> <mi>&sigma;</mi> <msub> <mi>B</mi> <mi>i</mi> </msub> </msub> <mo>+</mo> <msub> <mi>C</mi> <mn>3</mn> </msub> </mrow> </mfrac> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>4</mn> <mo>)</mo> </mrow> </mrow>

其中,分别是B1和Bi中所有像素灰度的平均值,分别是B1和Bi中所有像素灰度的方差,σ是B1和Bi之间像素灰度的协方差;

其中C1=(K1L)2,C2=(K2L)2,K1是常量0.01,K2是常量0.02,L是常量255;

然后,计算多尺度结构相似性;具体地,利用B1中各个像素的结构相似性计算p点的多尺度结构相似性MSSIMp

<mrow> <msub> <mi>MSSIM</mi> <mi>p</mi> </msub> <mo>=</mo> <mfrac> <mn>1</mn> <mi>N</mi> </mfrac> <munderover> <mo>&Sigma;</mo> <mrow> <mi>i</mi> <mo>=</mo> <mn>0</mn> </mrow> <mi>N</mi> </munderover> <mrow> <mo>(</mo> <mi>S</mi> <mi>S</mi> <mi>I</mi> <mi>M</mi> <mo>(</mo> <mrow> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> </mrow> <mo>)</mo> <mo>)</mo> </mrow> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>5</mn> <mo>)</mo> </mrow> </mrow>

其中N表示p点邻域彩色图像中像素的个数;

利用预设阈值ε,将空洞像素分为平滑区域空洞像素和非平滑区域空洞像素;若满足MSSIMp>ε,那么p为平滑区空洞像素;若MSSIM<ε,那么p为非平滑区空洞像素;

步骤2:对于深度图像中任意平滑区空洞进行填充

首先,利用最短距离算法对所有像素的深度进行聚类;对于深度图像中任意两个非空洞像素X和Y,如果它们满足公式(6),那么将它们聚为同一类;

|dX-dY|<T (6)

其中dX表示像素X的深度值,dY表示像素Y的深度值,T表示两像素间深度差异的阈值;空洞像素p的深度F利用公式(7)计算;

<mrow> <mi>F</mi> <mo>=</mo> <mfrac> <mn>1</mn> <mi>n</mi> </mfrac> <munderover> <mo>&Sigma;</mo> <mrow> <mi>t</mi> <mo>=</mo> <mn>0</mn> </mrow> <mi>n</mi> </munderover> <mrow> <mo>(</mo> <msub> <mi>E</mi> <mi>i</mi> </msub> <mo>&CenterDot;</mo> <mi>S</mi> <mi>S</mi> <mi>I</mi> <mi>M</mi> <mo>(</mo> <mrow> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> </mrow> <mo>)</mo> <mo>)</mo> </mrow> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>7</mn> <mo>)</mo> </mrow> </mrow>

Ei是有效像素qi的深度,n表示p点邻域中有效像素的个数;

步骤3:对于深度图像中任意非平滑区空洞进行填充

对于非平滑区域空洞像素点的填充,利用p点的邻域B1中有效像素qi进行求解,求解公式如下:

<mrow> <mi>F</mi> <mo>=</mo> <mfrac> <mrow> <munderover> <mo>&Sigma;</mo> <mrow> <mi>i</mi> <mo>=</mo> <mn>0</mn> </mrow> <mi>n</mi> </munderover> <mi>w</mi> <mrow> <mo>(</mo> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>&lsqb;</mo> <msub> <mi>E</mi> <mi>i</mi> </msub> <mo>+</mo> <mo>&dtri;</mo> <msub> <mi>E</mi> <mi>i</mi> </msub> <mo>&rsqb;</mo> </mrow> <mrow> <munderover> <mo>&Sigma;</mo> <mrow> <mi>i</mi> <mo>=</mo> <mn>0</mn> </mrow> <mi>n</mi> </munderover> <mi>w</mi> <mrow> <mo>(</mo> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> </mrow> </mfrac> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>8</mn> <mo>)</mo> </mrow> </mrow>

其中表示有效像素qi深度的梯度,w(p,qi)代表权重项,其计算为:

w(p,qi)=wd(p,qi)wc(p,qi)wm(p,qi) (9)

其中wd(p,qi),wc(p,qi)以及wm(p,qi)分别是距离权重项、颜色权重项和多尺度结构相似性权重项;它们的计算分别为:

<mrow> <msub> <mi>w</mi> <mi>d</mi> </msub> <mrow> <mo>(</mo> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>=</mo> <msup> <mi>e</mi> <mrow> <mo>(</mo> <mrow> <mo>-</mo> <mfrac> <mrow> <mo>|</mo> <mo>|</mo> <mi>x</mi> <mo>-</mo> <mi>y</mi> <mo>|</mo> <msup> <mo>|</mo> <mn>2</mn> </msup> </mrow> <mrow> <mn>2</mn> <msubsup> <mi>&sigma;</mi> <mi>d</mi> <mn>2</mn> </msubsup> </mrow> </mfrac> </mrow> <mo>)</mo> </mrow> </msup> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>10</mn> <mo>)</mo> </mrow> </mrow>

<mrow> <msub> <mi>w</mi> <mi>c</mi> </msub> <mrow> <mo>(</mo> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>=</mo> <msup> <mi>e</mi> <mrow> <mo>(</mo> <mrow> <mo>-</mo> <mfrac> <mrow> <mo>|</mo> <mo>|</mo> <msub> <mi>I</mi> <mi>p</mi> </msub> <mo>-</mo> <msub> <mi>I</mi> <msub> <mi>q</mi> <mi>i</mi> </msub> </msub> <mo>|</mo> <msup> <mo>|</mo> <mn>2</mn> </msup> </mrow> <mrow> <mn>2</mn> <msubsup> <mi>&sigma;</mi> <mi>c</mi> <mn>2</mn> </msubsup> </mrow> </mfrac> </mrow> <mo>)</mo> </mrow> </msup> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>11</mn> <mo>)</mo> </mrow> </mrow>

<mrow> <msub> <mi>w</mi> <mi>m</mi> </msub> <mrow> <mo>(</mo> <mi>p</mi> <mo>,</mo> <msub> <mi>q</mi> <mi>i</mi> </msub> <mo>)</mo> </mrow> <mo>=</mo> <msup> <mi>e</mi> <mrow> <mo>(</mo> <mrow> <mo>-</mo> <mfrac> <mrow> <mo>|</mo> <mo>|</mo> <msub> <mi>MSSIM</mi> <mi>p</mi> </msub> <mo>-</mo> <msub> <mi>MSSIM</mi> <msub> <mi>q</mi> <mi>i</mi> </msub> </msub> <mo>|</mo> <msup> <mo>|</mo> <mn>2</mn> </msup> </mrow> <mrow> <mn>2</mn> <msubsup> <mi>&sigma;</mi> <mi>M</mi> <mn>2</mn> </msubsup> </mrow> </mfrac> </mrow> <mo>)</mo> </mrow> </msup> <mo>-</mo> <mo>-</mo> <mo>-</mo> <mrow> <mo>(</mo> <mn>12</mn> <mo>)</mo> </mrow> </mrow>

其中符号||·||2表示2-范式;x表示p点的坐标,y表示qi点的坐标;表示当前像素p与邻域像素qi欧式距离的方差;表示像素p的邻域像素灰度的方差,是像素p的邻域像素多尺度结构相似性的方差;Ip分别表示彩色图像中像素p和像素qi的灰度值;表示qi点的多尺度结构相似性。

当前第2页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1