|
EDA365欢迎您登录!
您需要 登录 才可以下载或查看,没有帐号?注册
x
图像平滑处理
% Q" i ~$ j; K* p, V1 z* N" b9 q' i$ ^; t$ Y) [0 F1 p0 }7 t
目标& w5 }: U8 E+ P/ [4 G8 N' s* G
4 p# r8 \7 I# H2 |/ z* H# Y
本教程教您怎样使用各种线性滤波器对图像进行平滑处理,相关OpenCV函数如下: L7 o- F N& b7 c q: g
+ J" X: L) _3 h4 |4 T u; q( i- blur
- GaussianBlur
- medianBlur
- bilateralFilter( X: r$ y n5 \9 r
y7 b9 M8 A$ u$ P; g5 L
- z/ h0 j9 w' K8 {, E* O$ E( l) x
6 m2 J4 h7 t; T: K原理" b* t: z j# c% o& o8 m
Note 以下原理来源于Richard Szeliski 的著作 Computer Vision: Algorithms and Applications 以及 Learning OpenCV% K3 k& m+ Z$ x" r# `* L
平滑 也称 模糊, 是一项简单且使用频率很高的图像处理方法。
4 u$ ?; v0 s; F' t) @+ O
* M& P2 U! H6 L; ]) J% c0 L4 `- 平滑处理的用途有很多, 但是在本教程中我们仅仅关注它减少噪声的功用 (其他用途在以后的教程中会接触到)。
|3 h; Y4 s- U$ t
! L q4 X2 V( u( Y$ @: z, n6 H) ?% d: Y: g
- 平滑处理时需要用到一个 滤波器 。 最常用的滤波器是 线性 滤波器,线性滤波处理的输出像素值 (i.e.
) 是输入像素值 (i.e.
)的加权和 :2 e) r h2 Z5 d7 @
4 I; `% I' o' E& a" j+ `* B
, z [3 S2 G) W7 D* [
; Z1 F+ d6 t6 U/ s0 S2 v2 Y/ K" s/ S& o8 f6 t
称为 核, 它仅仅是一个加权系数。
% P( I, d: {: t. p8 r, k! F5 P# B0 e1 B: f& R8 ?1 Q
不妨把 滤波器 想象成一个包含加权系数的窗口,当使用这个滤波器平滑处理图像时,就把这个窗口滑过图像。$ _ f: i# @' ?. ^& P9 W( {- R
8 O8 T* M- c. L: J- 滤波器的种类有很多, 这里仅仅提及最常用的:
7 M% R+ X9 \9 c$ W) y' q ! ~9 H7 E3 g; |6 t$ d6 G: ~
1 V) j2 M4 y3 o& U% c
e9 t! I/ Q# y& m: j3 B
归一化块滤波器 (Normalized Box Filter)
4 X! b0 R. k' G+ R* S/ v% t6 D* s; T" L. y# Q6 j$ x3 a
最简单的滤波器, 输出像素值是核窗口内像素值的 均值 ( 所有像素加权系数相等)
$ S4 w: o2 b8 Y6 i4 }5 H9 N$ @
1 p/ e5 P' t- a1 {% t1 v' v0 O4 m核如下:
$ J0 |$ b) E( r6 j9 J& n z1 J
4 F" g+ T5 w( d y- K& V0 G
9 n$ C/ }0 d$ ^ d
8 ]9 y6 s$ d. i0 ~
0 K7 k+ j$ L' w9 e高斯滤波器 (Gaussian Filter)$ Q# _6 m: N6 v- u# x+ r+ ` }
9 H! \% |' O" e$ \1 Y6 r3 L
- 最有用的滤波器 (尽管不是最快的)。 高斯滤波是将输入数组的每一个像素点与 高斯内核 卷积将卷积和当作输出像素值。
& \. f: O- ~4 J$ B7 }( @8 [ 1 S& E) d. v- |) ^0 T
7 }. g _9 T6 l) \4 R- 还记得1维高斯函数的样子吗? t( ?3 ~9 O* r' n7 Z4 {# a
8 r9 c! j6 R7 Z+ O4 y
b3 c- t- r( }7 @+ @) t& g/ T
0 o7 v, |0 L% v: w: N' p
- m8 s! W7 j3 Y4 c) c
假设图像是1维的,那么观察上图,不难发现中间像素的加权系数是最大的, 周边像素的加权系数随着它们远离中间像素的距离增大而逐渐减小。: C2 I1 J& k0 w9 _( ?5 q
6 s3 i6 N) M2 ^+ L+ i4 }Note 2维高斯函数可以表达为 :5 u5 E( l& R1 \1 O; O
5 z7 o4 `& ]( z; Z K B' D+ O" M: Y4 V
9 \2 |: F% J% `/ f% `6 m) Q' z0 O" b" s3 z) E0 a1 n. \
其中
为均值 (峰值对应位置),
代表标准差 (变量
和变量
各有一个均值,也各有一个标准差)
& f) D+ j' L' {2 \
2 H7 e! x- D; ~* Y% A' e$ M( Q
1 I( K9 W# {5 @/ M中值滤波器 (Median Filter)) @/ C4 B$ T6 ?( d5 ~
8 U$ U+ i4 {1 X7 `' Y! R
中值滤波将图像的每个像素用邻域 (以当前像素为中心的正方形区域)像素的 中值 代替 。
, M6 _& h& r- Q3 ?9 ]& L+ `1 L0 \# J9 W+ V! C5 c& Y" G
% \5 ?4 R- d: e9 E9 z' D8 c4 R7 z
双边滤波 (Bilateral Filter)9 w. D: n$ B% h' {1 J
- R- V+ G1 M& e' f- 目前我们了解的滤波器都是为了 平滑 图像, 问题是有些时候这些滤波器不仅仅削弱了噪声, 连带着把边缘也给磨掉了。 为避免这样的情形 (至少在一定程度上 ), 我们可以使用双边滤波。
- 类似于高斯滤波器,双边滤波器也给每一个邻域像素分配一个加权系数。 这些加权系数包含两个部分, 第一部分加权方式与高斯滤波一样,第二部分的权重则取决于该邻域像素与当前像素的灰度差值。0 R8 y, s+ w3 d. U5 D7 t/ C
$ n/ Q- S* b5 s* U5 ^% @7 Q! D8 Q- p
0 P( ?& L8 @, P0 b
4 @& Z0 {) O- r源码 V& d* o/ J, ?, i( N: H
3 ?. @0 i" ^+ x# Z6 k3 e本程序做什么?
6 M8 `1 O" y2 b
( d% Z0 ~$ A0 K( ^: J0 Y- 装载一张图像
- 使用4种不同滤波器 (见原理部分) 并显示平滑图像$ G' d5 e0 i; J. l) z; b# R* J
" e! Z8 A5 ~4 f' G3 H6 u# a( @7 u6 l) u6 B* U! l2 E1 k. ~
代码一瞥:" d& R" Q! T) b8 a7 S. D
3 F0 A) g/ d8 U" ^1 W% z) q
#include "opencv2/imgproc/imgproc.hpp"
5 o' ?6 W. Z4 E4 z#include "opencv2/highgui/highgui.hpp"7 G6 |. j3 D! ~# Y+ `
) o x6 Q# P9 t5 Q9 B- u" j6 X) f
using namespace std;* N" i2 c% y* v3 c
using namespace cv;4 g) G8 K. V" K- W
! Y, [- i% E3 \& V1 Y6 E' q/// 全局变量+ w! Q2 P [6 G9 Z4 _, l- v
int DELAY_CAPTION = 1500;9 { M5 d; g: t5 C
int DELAY_BLUR = 100;
/ W) z+ q' S9 y2 _' \: @* x! ^int MAX_KERNEL_LENGTH = 31;
2 L: q8 Y% p, ]6 N) `; l+ p e- B
Mat src; Mat dst;* _) a* K8 } d* @* v
char window_name[] = "Filter Demo 1";$ _% f( s9 S4 ~6 w
$ F8 t6 ~; x* v0 [/ t% B
/// 函数申明' v+ C8 \' i' Q+ T4 V0 g- x6 z) X: n( J
int display_caption( char* caption );
2 j& ^) u( G W. w; hint display_dst( int delay );) i9 y1 z. \7 |- ]
$ l; O2 \0 ~6 ]* L( n5 K' g0 m3 j
/**, O# o5 ?( T' T1 B' n! F
* main 函数. L7 ]. s, s8 h+ b, t# E. B5 L! `
*/
8 G+ @6 V6 f6 y& p2 j3 ~0 U/ h( i int main( int argc, char** argv )5 U5 E% k7 [' P% ?4 U! x
{
a* R" Y/ y8 K. G8 Y namedWindow( window_name, CV_WINDOW_AUTOSIZE );
" ?+ e7 K2 [ x3 D& @, A8 x0 S8 @9 F r2 @3 O- f' a3 R
/// 载入原图像- z' Z: L4 J/ L( f+ z6 E0 `2 r" }
src = imread( "../images/lena.jpg", 1 ); ~ R/ q/ g; s0 V4 i9 @4 g
2 b. i, y% W- O if( display_caption( "Original Image" ) != 0 ) { return 0; }
8 P' U+ u$ c2 o5 T3 _9 f1 Y5 t' Z
dst = src.clone();7 @% _ K+ A& }- l/ n
if( display_dst( DELAY_CAPTION ) != 0 ) { return 0; }- r3 ^3 A, g c3 i) e
6 U$ @- J8 E/ ?% c
/// 使用 均值平滑 e" u9 \' U8 k; g" H Y! v
if( display_caption( "Homogeneous Blur" ) != 0 ) { return 0; }/ R! z; H O0 s. E1 s
; i P; ~+ X) d1 y0 t for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
; a% g; R1 ?# ^ { blur( src, dst, Size( i, i ), Point(-1,-1) );; {7 Z: E& M: T) F7 R
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
3 n/ M# s7 A! u* P6 b/ v& \
# P+ u) Q) [ T$ r9 s /// 使用高斯平滑
5 d' Z5 u5 p1 E' m. F3 P3 H9 { if( display_caption( "Gaussian Blur" ) != 0 ) { return 0; }
; I9 g0 }$ k3 v' U) J0 S) v
# L3 I H; D: G; \8 Q for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 ), {+ e1 K. o9 Y0 ?% w7 E
{ GaussianBlur( src, dst, Size( i, i ), 0, 0 );9 U6 S3 g6 `) M: P* k* M6 d% }
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
% R9 e/ K* Q+ k. c0 S ^; ~" ]4 p0 d
/// 使用中值平滑
8 ^; E* K( F/ R if( display_caption( "Median Blur" ) != 0 ) { return 0; }+ n- B$ r: C2 ~9 [
- s$ _/ O: V% f8 d1 g3 |8 H2 O1 }' b
for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )$ V& Q9 a/ U- v8 ^* E; l( n
{ medianBlur ( src, dst, i );& D6 u9 j' p/ D1 P
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
) s# s, r4 S- ?7 ?- z3 D
. b1 o% u! t$ E# g /// 使用双边平滑
' U2 ]& P% f# B; Z if( display_caption( "Bilateral Blur" ) != 0 ) { return 0; }, \# l; F8 d: \8 _. j" l
. H# [4 i- p( X9 e8 @6 K7 T
for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
P0 `" p0 o- z( L/ d- l# K { bilateralFilter ( src, dst, i, i*2, i/2 );: k5 Y7 |" @6 d8 ~: Q5 a
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }" Q+ a {; r2 {% q: o, f# \
$ s; F4 N" E E4 o) R# s5 j B" i /// 等待用户输入
2 p" L! e9 Q) ?. ^$ l, y8 L display_caption( "End: Press a key!" );: Q* B8 v7 N& Y5 `: d2 g
9 Q6 J4 P# l9 B+ V' M
waitKey(0);+ D$ x" [+ a: [+ _- L0 h
return 0;
0 v6 a8 S% R: t z% q/ l% X }+ U* O& @: w3 a; W* p, ]
% D5 w" D* }, y$ q Z0 ]; \
int display_caption( char* caption )# B% p/ [% f# R: Z( }, {$ T5 X
{
& I/ P* K' e4 G% C dst = Mat::zeros( src.size(), src.type() );
/ X; E0 @, B5 @6 s. v putText( dst, caption,8 x c/ f' D# v! H6 B
Point( src.cols/4, src.rows/2),0 H& r0 U5 [% G5 J* c( B$ q4 k
CV_FONT_HERSHEY_COMPLEX, 1, Scalar(255, 255, 255) );% x$ M5 V3 A6 W' ^1 T8 k, M" ?' @
s' ~) D' S) V0 e
imshow( window_name, dst );; T% p. P6 W6 |4 @" K3 ~% _7 p
int c = waitKey( DELAY_CAPTION );
' |( k. l3 M: A( j9 z- e7 { b if( c >= 0 ) { return -1; }; E2 Z4 m- _ y1 W
return 0;
$ L3 L5 g- F- b9 l$ r9 n2 V }
2 C1 Q5 ^! ]* i: f8 f
! v( {- [6 W! o' Q7 H/ s* N int display_dst( int delay )7 U5 Z) v: a) s0 s; [
{
3 ]- Y; E# }' r. {6 l imshow( window_name, dst );
* j7 X! u2 i W6 _3 k int c = waitKey ( delay );
7 ]7 j3 p2 {. \4 u, Y if( c >= 0 ) { return -1; }& r! O7 @$ ?4 w! Q8 `; L+ Q
return 0;/ H( d- U0 Y& l4 B, t
}
7 f) y) u: J! p0 D: x# H8 V# q7 z
3 y$ i* z; v7 i/ m. ]& G g9 S9 Q4 @7 p& i1 o1 S: J9 z7 X
解释
' A7 Y2 k7 X! ?/ b3 H1 u/ U8 t' H& W! ^% E4 M6 e6 A
- 下面看一看有关平滑的OpenCV函数,其余部分大家已经很熟了。
- 归一化块滤波器:3 K" T7 s: S5 G/ ^9 n0 e; ]
; r5 i9 O9 v1 z: w/ G5 a1 U
5 c- N- G. q7 A0 v; u
OpenCV函数 blur 执行了归一化块平滑操作。
- z' P# y' T) _. N( ~$ c4 L; u: L( F6 G6 w0 H+ a
for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )# r; k& | B" ~! J, y
{ blur( src, dst, Size( i, i ), Point(-1,-1) );
6 j# k$ ]& V$ ]" O8 A) h if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }& K3 r5 p/ e8 ~
7 K& q/ y ~5 K+ e' x我们输入4个实参 (详细的解释请参考 Reference):
0 F- q4 [7 ?; `2 E# n5 l5 V
5 {( A" f: H7 \5 M- src: 输入图像
- dst: 输出图像
- Size( w,h ): 定义内核大小( w 像素宽度, h 像素高度)
- Point(-1, -1): 指定锚点位置(被平滑点), 如果是负值,取核的中心为锚点。4 u+ u; a6 w- Q! J1 z
3 [6 V/ I. L( u) R
+ h7 y0 c% k; K* D% F 3. 高斯滤波器:
X0 f( k) r4 x6 j+ u
4 u# w/ v- Y, i) JOpenCV函数 GaussianBlur 执行高斯平滑 :! K. h% H i! r% h+ O
. f' \3 [1 h8 Z: Y. Gfor ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )) F# q, X4 u" p) r# c
{ GaussianBlur( src, dst, Size( i, i ), 0, 0 );8 c1 D% M3 l. i( G& v- G
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
: p% n" _9 B+ D/ w- @8 {
T% {3 K2 O) \5 h我们输入4个实参 (详细的解释请参考 Reference):: s- @! Q ]. G' ?9 T/ T4 X. @, P! U u
+ c9 ?; L6 s+ d' K: s' ?- src: 输入图像
- dst: 输出图像
- Size(w, h): 定义内核的大小(需要考虑的邻域范围)。
和
必须是正奇数,否则将使用
和
参数来计算内核大小。 -
: x 方向标准方差, 如果是
则
使用内核大小计算得到。 -
: y 方向标准方差, 如果是
则
使用内核大小计算得到。.6 s" o3 T. n! `7 O N: g
% j, f( x( n6 l7 A' j, G
0 P1 \' O6 j1 A; J5 c& [: s+ \
2 o# k& I1 ?; s% u( q2 c9 W中值滤波器:
& q a- z7 D% U8 o2 g* S
+ h: n0 N+ B9 UOpenCV函数 medianBlur 执行中值滤波操作:
' B# D) K' ^) m4 p% ~: m) p: y
for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
5 ^; R% o0 {$ X: G( \ V% R) o { medianBlur ( src, dst, i );/ P4 n. C/ c4 Q0 i" n
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
1 _ x) C: D! V6 L. [, Q2 U7 V
) M3 \/ @! e/ Z9 ^! y! Y8 S, l* }我们用了3个参数:7 \/ S' C# w* m1 \: Y1 W0 D7 M! X
# E( z/ X, A: jsrc: 输入图像
' c# t" d" Q+ d- hdst: 输出图像, 必须与 src 相同类型
' A7 L0 M7 i1 q/ c& [% Xi: 内核大小 (只需一个值,因为我们使用正方形窗口),必须为奇数。
6 s d3 b8 k2 s# v2 m0 M% w! ]2 g. ~" P7 v' x @$ N$ Y
) m: Q. _* n$ c' O% `
双边滤波器; c6 I- d/ D" s$ z% D- x: s
6 m# G2 |% G: o Y9 S
OpenCV函数 bilateralFilter 执行双边滤波操作:
* p) T2 d7 L4 z" R( O% N; C( C4 O8 G: D8 `, ?9 c' J- L: z
for ( int i = 1; i < MAX_KERNEL_LENGTH; i = i + 2 )
1 Q2 R+ u1 ], l' ^% \9 I { bilateralFilter ( src, dst, i, i*2, i/2 );% Y$ ?. n4 A6 x }6 l1 l2 X4 O! F, E
if( display_dst( DELAY_BLUR ) != 0 ) { return 0; } }
R0 }# |% s4 @* n4 D: ~2 y
. u' L% n8 a4 @1 L# `3 u6 ^& r& V我们使用了5个参数:5 n8 F# S2 Z! x3 S
$ ^& s3 ?1 j' }2 W8 S( z* H# [: M
- src: 输入图像
- dst: 输出图像
- d: 像素的邻域直径
-
: 颜色空间的标准方差 -
: 坐标空间的标准方差(像素单位)4 l0 C$ q6 U; v& M) N3 T
* u4 i- O5 d* ]
: q4 O' x8 w4 Z- i6 z9 ^) H8 B- D. g# Q$ X
结果& G5 I1 f5 L- d* F3 Q+ Z" p+ j
6 c' B6 x0 \: ~8 y
- 程序显示了原始图像( lena.jpg) 和使用4种滤波器之后的效果图。
( A" B% p+ A/ U" l$ J ) F- B7 F2 Y0 W- j9 M
" ^9 H F0 _5 {6 l- 这里显示的是使用 中值滤波 之后的效果图:. w9 ]+ _3 I8 O7 k6 Y6 n$ {
/ N( L' V+ h' Q J! c3 o; @1 f7 }1 y. e z1 o5 |9 i
: A8 }1 W8 O9 @1. 简介5 F. m. m# w: B# a
: O) Z. m9 ~; C2 y2 P图像平滑是一个重要的操作,而且有多种成熟的算法。这里主要简单介绍一下Bilateral方法(双边滤波),这主要是由于前段时间做了SSAO,需要用bilateral blur 算法进行降噪。Bilateral blur相对于传统的高斯blur来说很重要的一个特性即可可以保持边缘(Edge Perseving),这个特点对于一些图像模糊来说很有用。一般的高斯模糊在进行采样时主要考虑了像素间的空间距离关系,但是却并没有考虑像素值之间的相似程度,因此这样我们得到的模糊结果通常是整张图片一团模糊。Bilateral blur的改进就在于在采样时不仅考虑像素在空间距离上的关系,同时加入了像素间的相似程度考虑,因而可以保持原始图像的大体分块进而保持边缘。在于游戏引擎的post blur算法中,bilateral blur常常被用到,比如对SSAO的降噪。
: Y; p7 j7 w. i" E5 S+ ^3 B
1 h U# J% i) ?7 C; f* T2. 原理 V) P9 z2 |" u3 }( J F) e* A
6 a) X' z1 X7 a
滤波算法中,目标点上的像素值通常是由其所在位置上的周围的一个小局部邻居像素的值所决定。在2D高斯滤波中的具体实现就是对周围的一定范围内的像素值分别赋以不同的高斯权重值,并在加权平均后得到当前点的最终结果。而这里的高斯权重因子是利用两个像素之间的空间距离(在图像中为2D)关系来生成。通过高斯分布的曲线可以发现,离目标像素越近的点对最终结果的贡献越大,反之则越小。其公式化的描述一般如下所述:
! `) d4 x( e' d! M% _/ ~5 w, w7 x- d! Q+ q0 w$ c/ ~
+ F4 z2 i3 @" a/ y/ |
" W# x4 u1 |7 h6 v3 X" z7 \
- f% D' p& i8 M) X3 H; o
. @+ f; Y' Z$ S) K: G: X
其中的c即为基于空间距离的高斯权重,而
用来对结果进行单位化。8 J* I7 `$ N4 ?) r( {- O9 D* \
; ?, n9 c' U& Q. e高斯滤波在低通滤波算法中有不错的表现,但是其却有另外一个问题,那就是只考虑了像素间的空间位置上的关系,因此滤波的结果会丢失边缘的信息。这里的边缘主要是指图像中主要的不同颜色区域(比如蓝色的天空,黑色的头发等),而Bilateral就是在Gaussian blur中加入了另外的一个权重分部来解决这一问题。Bilateral滤波中对于边缘的保持通过下述表达式来实现:; I7 w; D0 `' Y) N9 k% n
" [) j3 G7 v5 c8 q* H% C
, `' u# U3 q) ` B. A: R: \
0 E* G$ t4 [" S6 K
0 ?5 D1 e5 c4 \1 w0 q! {
' s& d: Q- A3 l, {; k- o
其中的s为基于像素间相似程度的高斯权重,
同样用来对结果进行单位化。对两者进行结合即可以得到基于空间距离、相似程度综合考量的Bilateral滤波:! ~- t2 Z6 D/ N0 L) h
% G7 A8 b" C/ F5 J5 q% _. G% k
6 ]* E, B+ F# |5 M% }# Q
7 b- g \ m- n( A/ b" b
) S' ]' v; p/ U$ H- z" c K/ ^& T2 }' q9 A. X
上式中的单位化分部
综合了两种高斯权重于一起而得到,其中的c与s计算可以详细描述如下:& K$ ?' J# @6 o& Q1 @
8 N2 n. y4 p" ?2 [! x
. m6 r' T; Q3 c" {
% y) `; e/ k5 B$ Q3 ]+ g( X: c( B" W且有
" g$ ^8 F8 ?- u
# z9 i5 }# ?; ?' H, g
% }2 K |; W, @
9 U/ T" c9 I* V且有
2 s' e2 ]" x0 f! s& u5 c" G; ~2 E4 T S, t- ]' K
上述给出的表达式均是在空间上的无限积分,而在像素化的图像中当然无法这么做,而且也没必要如此做,因而在使用前需要对其进行离散化。而且也不需要对于每个局部像素从整张图像上进行加权操作,距离超过一定程度的像素实际上对当前的目标像素影响很小,可以忽略的。限定局部子区域后的离散化公就可以简化为如下形式:
$ \) y8 K; c- T) Z. T% e5 J; ]# H; k, ]# _( g ~
* b5 z, ~) f5 ~! d7 P/ ~
' x+ C+ p" |2 h z3 A! F3 N上述理论公式就构成了Bilateral滤波实现的基础。为了直观地了解高斯滤波与双边滤波的区别,我们可以从下列图示中看出依据。假设目标源图像为下述左右区域分明的带有噪声的图像(由程序自动生成),蓝色框的中心即为目标像素所在的位置,那么当前像素处所对应的高斯权重与双边权重因子3D可视化后的形状如后边两图所示:
. P* o- W, z/ |# A" t5 {: D# j _7 F: }+ k
7 K. }1 q# Q6 d: J( N; q
" z" W6 i" l3 }) _8 O; ^
; p% o& Z- Z- B
6 _4 `2 N) l# U s5 ?
/ i# {" H1 s5 t
/ w& O* X: T% t/ G
左图为原始的噪声图像;中间为高斯采样的权重;右图为Bilateral采样的权重。从图中可以看出Bilateral加入了相似程度分部以后可以将源图像左侧那些跟当前像素差值过大的点给滤去,这样就很好地保持了边缘。为了更加形象地观察两者间的区别,使用Matlab将该图在两种不同方式下的高度图3D绘制出来,如下:7 R9 T' \, P/ ~* H2 i2 g$ k0 G# ]
4 W* H( \7 t7 U" N/ d
, R! a5 p! v5 R) m- m9 n
! v# O. d! i& G2 k8 n1 o# [
- A' `* {- H. k' ~3 r) }4 y
% n' n3 R( n' o0 y! w
* c5 A& i8 y* c: I& @: ]* O& u! F3 T u
- b% f( }; e6 ~3 o上述三图从左到右依次为:双边滤波,原始图像,高斯滤波。从高度图中可以明显看出Bilateral和Gaussian两种方法的区别,前者较好地保持了边缘处的梯度,而在高斯滤波中,由于其在边缘处的变化是线性的,因而就使用连累的梯度呈现出渐变的状态,而这表现在图像中的话就是边界的丢失(图像的示例可见于后述)。 7 g9 M" E) K' c- O; a1 X
4 {7 a8 x8 D+ w) {, o/ p/ B9 {
3. 代码实现
, H% V1 u( e* w# W$ l7 l
2 }* k' Q' C6 w$ R& C3 C& N有了上述理论以后实现Bilateral Filter就比较简单了,其实它也与普通的Gaussian Blur没有太大的区别。这里主要包括3部分的操作: 基于空间距离的权重因子生成;基于相似度的权重因子的生成;最终filter颜色的计算。( s- U+ K( g4 b& p7 N" v
* t; C2 Z/ S% L: j
3.1 Spatial Weight% v3 H9 J N* d! ]4 h2 y% B
1 ?0 k. ~( z. e这就是通常的Gaussian Blur中使用的计算高斯权重的方法,其主要通过两个pixel之间的距离并使用如下公式计算而来:
* M+ D7 Q8 Q5 z4 C/ k. [7 P0 O$ W' k( d! q
9 C/ t1 P2 n- N3 J7 h$ h
$ @) d9 E9 I' b3 O( o4 G其中的
就表示两个像素间的距离,比如当前像素与其右边紧邻的一个像素之间的距离我们就可以用
来计算,也即两个二维向量{0 , 0}以及{0 , 1}之间的欧氏距离。直接计算一个区域上的高斯权重并单位化后就可以进行高斯模糊了。
" b" C+ I3 r2 O" c+ y: A9 s
! T9 y# v# n% A& i& v0 O: w' @" W/ W
6 J% }- a0 _% M e M/ l3.2 Similarity Weight
2 ^$ s% p1 Q2 f
- o% V: k9 H. E- V8 T与基于距离的高斯权重计算类似,只不过此处不再根据两个pixel之间的空间距离,而是根据其相似程度(或者两个pixel的值之间的距离)。- b7 `: B1 S7 J6 d0 H: e3 L4 [
9 L }4 T$ X5 P. P! }0 r8 i; w" R2 t
% W2 a& |$ @4 ~9 k2 O D7 w
: {/ y7 A- W/ Y3 ]& ?* P
其中的
表示两个像素值之间的距离,可以直接使用其灰度值之间的差值或者RGB向量之间的欧氏距离。1 d. g+ C+ Q5 n; x+ i
) D. l, Q, F7 _; `! v& q. [0 R* D0 ?! Y/ w3 _* s: R- P
3.3 Color Filtering
$ Z5 p- H; X; L
8 J, q- W" g3 W" @2 u. J+ S有了上述两部分所必需的权重因子之后,那么具体的双边滤波的实现即与普通的高斯滤波无异。主要部分代码如下述:% ]- p( t/ v8 F4 z0 e
+ }$ y$ c. d- `0 l, {1 a& J/ r
- UCHAR3 BBColor(int posX , int posY)
- {
- int centerItemIndex = posY * picWidth4 + posX * 3 , neighbourItemIndex;
- int weightIndex;
- double gsAccumWeight = 0;
- double accumColor = 0;
- // 计算各个采样点处的Gaussian权重,包括closeness,similarity
- for(int i = -number ; i <= number ; ++i)
- {
- for(int j = -number ; j <= number ; ++j)
- {
- weightIndex = (i + number) * (number * 2 + 1) + (j + number);
- neighbourItemIndex = min(noiseImageHeight - 1 , max(0 , posY + j * radius)) * picWidth4 +
- min(noiseImageWidth - 1 , max(0 , posX + i * radius)) * 3;
- pCSWeight[weightIndex] = LookupGSWeightTable(pSrcDataBuffer[neighbourItemIndex] , pSrcDataBuffer[centerItemIndex]);
- pCSWeight[weightIndex] = pGSWeight[weightIndex] * pGCWeight[weightIndex];
- gsAccumWeight += pCSWeight[weightIndex];
- }
- }
- // 单位化权重因子
- gsAccumWeight = 1 / gsAccumWeight;
- for(int i = -number ; i <= number ; ++i)
- {
- for(int j = -number ; j <= number ; ++j)
- {
- weightIndex = (i + number) * (number * 2 + 1) + (j + number);
- pCSWeight[weightIndex] *= gsAccumWeight;
- }
- }
- // 计算最终的颜色并返回
- for(int i = -number ; i <= number ; ++i)
- {
- for(int j = -number ; j <= number ; ++j)
- {
- weightIndex = (i + number) * (number * 2 + 1) + (j + number);
- neighbourItemIndex = min(noiseImageHeight - 1 , max(0 , posY + j * radius)) * picWidth4 +
- min(noiseImageWidth - 1 , max(0 , posX + i * radius)) * 3;
- accumColor += pSrcDataBuffer[neighbourItemIndex + 0] * pCSWeight[weightIndex];
- }
- }
- return UCHAR3(accumColor , accumColor , accumColor);
- }
* u6 `+ a: B* j2 }& Y8 E 2 \6 @, s8 w# T9 T( v
其中的相似度分部的权重s主要根据两个Pixel之间的颜色差值计算面来。对于灰度图而言,这个差值的范围是可以预知的,即[-255, 255],因而为了提高计算的效率我们可以将该部分权重因子预计算生成并存表,在使用时快速查询即可。使用上述实现的算法对几张带有噪声的图像进行滤波后的结果如下所示:, P/ R4 m3 I3 J# x+ w/ }9 q
, K( z! V E: B6 m: J' d* u
X; m" z$ D: R( h: I! n
* D/ v9 f z9 [; D
L- ]7 `) O& K& h$ ?% r* ?2 Z+ Z. T8 a5 {4 W
上图从左到右分别为:双边滤波;原始图像;高斯滤波。从图片中可以较为明显地看出两种算法的区别,最直观的感受差别就是使用高斯算法后整张图片都是一团模糊的状态;而双边滤波则可以较好地保持原始图像中的区域信息,看起来仍然嘴是嘴、眼是眼(特别是在第一张美女图像上的效果!看来PS是灰常重要啊~~^o^)。
- p! E' V4 w! y$ y l9 k6 c, `) m6 t5 `! D, r3 f/ g5 s
0 Z( U- w0 Y% u4. 在SSAO中的使用
. y3 f7 S+ z' }$ c+ Q) X
: X( `! T6 k" Q" z( Y- n在上述实现中的边缘判定函数主要是通过两个像素值之间的差异来决定,这也是我们观察普通图片的一种普遍感知方式。当然,也可以根据使用的需求情况来使用其它的方式判断其它定义下的边缘,比如使用场景的depth或是normal。比如在对SSAO进行滤波时可以直接使用Depth值来行边缘判断。首先,设置一个深度的阈值,在作边缘检测时比较两点间的depth差值,如果差值大于阈值,则认为其属于不同的区域,则此处就应为边界。使用此方法得到的效果可见于下图所示:
' H9 c6 L4 L6 E0 ]1 D$ \9 x0 u1 x" o4 h( s+ H" U# R m+ Y. w9 J
. _4 y7 a+ u; u* U2 R9 v
6 _8 T" J! W. j' T高斯滤波
6 D2 i6 U* q1 m, g1 J. }& H6 [8 `* g# F$ w7 L* U! v
! G% V1 W4 j- E) e
9 C5 ^4 p5 s( O
双边滤波6 N T( E) s4 Z% K3 G5 q. Q
4 {5 {7 I4 Y3 \3 O
在得到滤波之后的SSAO图像之后,与原始图像进行直接的整合就可以得到最终的渲染效果,如下图所示:& p& V7 K5 {6 D/ R' u/ x/ O0 p
) N4 g3 M9 G3 t* g: V
9 m! s1 U F& T' V9 r9 W: S
. X$ g1 @. W$ zSSAO关闭
3 B" d! C& O( M3 t
1 Q" G. A7 e' D( ?1 L$ E
1 V+ ]: p: U/ g3 Z! w3 ~( _$ x* h! D* j8 q
SSAO开启
5 U8 F& q5 K3 V. y3 y! t8 u
6 Q# q. ~' L! t6 D1 n! R1 I l4 x7 y7 ? y0 [
1 G; n: e" |7 a. Y {* q; ?" i
|
|