opencv_python学习笔记十七

20
图像金字塔

高斯金字塔

拉普拉斯金字塔

所用函数:

def
pyrDown(src,
#输入图像

dst=None,
#输出图像

dstsize=None,
#输出图像的大小

borderType=None)#图像边界的处理方式

def
pyrUp(src,

dst=None,

dstsize=None,

borderType=None)

1
原理

示例代码如下:

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# @Time : 2016/11/21 14:01

# @Author : Retacn

# @Site :
图像金字塔

# @File : imagePyr.py

# @Software: PyCharm

import
cv2

import
numpy
as
np,sys

from
matplotlib
import
pyplot
as
plt

img=cv2.imread('../test.jpg',0)

lower_reso=cv2.pyrDown(img)

higher_reso=cv2.pyrUp(img)

plt.subplot(131),plt.imshow(img,'gray')

plt.title('Original')

#,plt.xticks([]),plt.yticks([])

plt.subplot(132),plt.imshow(lower_reso,'gray')

plt.title('Lower_reso')

#,plt.xticks([]),plt.yticks([])

plt.subplot(133),plt.imshow(higher_reso,'gray')

plt.title('Higher_reso')

#,plt.xticks([]),plt.yticks([])

plt.show()

2
使用金字塔进行图像融合

示例代码如下:

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# @Time : 2016/11/21 14:44

# @Author : Retacn

# @Site :
图像融合

# @File : ImageFusion.py

# @Software: PyCharm

import
cv2

import
numpy
as
np,sys

from
matplotlib
import
pyplot
as
plt

"""

使得img1的大小与img2相同

"""

def
sameSize(img1, img2):

rows, cols, dpt = img2.shape

dst = img1[:rows,:cols]

return
dst

#读入图像

apple=cv2.imread('../apple.jpg')

orange=cv2.imread('../orange.jpg')

#为图像苹果创建一个高斯金字塔

temp_c=apple.copy()

apple_gp=[temp_c]

for
i
in
range(6):

temp_c=cv2.pyrDown(temp_c)

apple_gp.append(temp_c)

#为图像橙子创建一个高斯金字塔

temp_c=orange.copy()

orange_gp=[temp_c]

for
i
in
range(6):

temp_c=cv2.pyrDown(temp_c)

orange_gp.append(temp_c)

#为图像苹果创建一个拉普拉斯金字塔

apple_lp=[apple_gp[5]]

for
i
in
range(5,0,-1):

GE=cv2.pyrUp(apple_gp[i])

L=cv2.subtract(apple_gp[i-1],sameSize(GE,apple_gp[i-1]))

apple_lp.append(L)

#为图像橙子创建一个拉普拉斯金字塔

orange_lp=[orange_gp[5]]

for
i
in
range(5,0,-1):

GE=cv2.pyrUp(orange_gp[i])

L=cv2.subtract(orange_gp[i-1],sameSize(GE,orange_gp[i-1]))

orange_lp.append(L)

#1/2 Laplace金字塔进行拼接

LS=[]

for
apple_l,orange_l
in
zip(apple_lp,orange_lp):

rows,cols,dpt=apple_l.shape

ls = np.hstack((apple_l[:,0:cols/2],
orange_l[:,cols/2:]))

LS.append(ls)

#对拼接后的laplace金字塔重建,取得融合后的图像

ls_=LS[0]

for
i
in
range(1,6):

ls_=cv2.pyrUp(ls_)

ls_=cv2.add(sameSize(ls_,LS[i]),LS[i])

real=np.hstack((apple[:,:cols/2],orange[:,cols/2:]))

#保存图像

#cv2.imwrite('Pyramid_blending2.jpg',ls)

#cv2.imwrite('Direct_blending.jpg',real)

#显示图像

plt.subplot(121),plt.imshow(ls_,'gray')

plt.title('ls_')

plt.subplot(122),plt.imshow(real,'gray')

plt.title('real')

plt.show()