python+opencv图像金字塔融合

本文参考《OpenCV-Python 中文教程》上的图像融合内容,在测试代码时运行有两种错误,

第一个问题:应该是python版本问题

第二个问题:教程上的代码是做六层高斯金子塔,然后根据高斯金子塔在做拉普拉斯金字塔,在这过程中有图片在某一层不能被二整除,导致维度不一致,出现错误。所以在做融合时图片在每一层都能被二整除

我的解决方案:

1.通过改变代码,使代码能够在python3.5上运行。

2.通过改变代码,使被融合的图片不必每层都要被2整除

代码如下:

import cv2
import numpy as np,sys

A = cv2.imread('./data/apple.jpg')
B = cv2.imread('./data/orange.jpg')
print(type(A))
# generate Gaussian pyramid for A
G = A.copy()
print(G)
gpA = [G]
for i in range(6):
    G = cv2.pyrDown(G)
    gpA.append(G)
G = B.copy()
gpB = [G]
for i in range(6):
    G = cv2.pyrDown(G)
    gpB.append(G)
lpA = [gpA[5]]
for i in range(6,0,-1):
    print(i)
    GE = cv2.pyrUp(gpA[i])
    GE=cv2.resize(GE,gpA[i - 1].shape[-2::-1])
    L = cv2.subtract(gpA[i-1],GE)
    print(L.shape)
    lpA.append(L)
# generate Laplacian Pyramid for B
lpB = [gpB[5]]
for i in range(6,0,-1):
    print(i)
    GE = cv2.pyrUp(gpB[i])
    GE = cv2.resize(GE, gpB[i - 1].shape[-2::-1])
    L = cv2.subtract(gpB[i-1],GE)
    print(L.shape)
    lpB.append(L)
# Now add left and right halves of images in each level
LS = []
lpAc=[]
for i in range(len(lpA)):
    b=cv2.resize(lpA[i],lpB[i].shape[-2::-1])
    print(b.shape)
    lpAc.append(b)
print(len(lpAc))
print(len(lpB))
j=0
for i in zip(lpAc,lpB):
    print(i)
    print('ss')
    la,lb=i
    print(la)
    print(lb)
    rows,cols,dpt = la.shape
    ls = np.hstack((la[:,0:cols//2], lb[:,cols//2:]))
    j=j+1
    print(j)
    LS.append(ls)
ls_ = LS[0]
for i in range(1,6):
    ls_ = cv2.pyrUp(ls_)
    ls_= cv2.resize(ls_, LS[i].shape[-2::-1])
    ls_ = cv2.add(ls_, LS[i])
# image with direct connecting each half
B= cv2.resize(B, A.shape[-2::-1])
real = np.hstack((A[:,:cols//2],B[:,cols//2:]))
cv2.imwrite('Pyramid_blending2.jpg',ls_)
cv2.imwrite('Direct_blending.jpg',real)

被融合图片:

直接融合后结果:

图像金子塔融合后的结果:

猜你喜欢

转载自blog.csdn.net/qq_29023939/article/details/81101168