本文参考《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)
被融合图片:
直接融合后结果:
图像金子塔融合后的结果: