图像处理：使用金字塔进行图像无缝融合(OpenCV官网代码的纠正)

一 问题介绍

图像金字塔的一个应用是图像融合。例如，在图像缝合中，你需要将两幅图叠在一起，但是由于连接区域图像像素的不连续性，整幅图的效果看起来会很差。这时图像金字塔就可以排上用场了，它可以帮你实现无缝连接。这里的一个经典案例就是将两个水果融合成一个，如下图所示：

二 实现步骤

1. 读入两幅图像，苹果和橘子
2. 构建苹果和橘子的高斯金字塔（5 层）
3. 根据高斯金字塔计算拉普拉斯金字塔
4. 在拉普拉斯的每一层进行图像融合（苹果的左边与橘子的右边融合）
5. 根据融合后的图像金字塔重建原始图像。

三 代码实现

# -*- coding: utf-8 -*-
from cv2 import cv2
import numpy as np
A = cv2.imread('apple.png')
#注意：为了使后面可以逐渐减半，这里的尺寸必须为2的次幂
A = cv2.resize(A,(256,256),interpolation=cv2.INTER_CUBIC) 
B = cv2.imread('orange.png')
B = cv2.resize(B,(256,256),interpolation=cv2.INTER_CUBIC)
# 生成高斯金字塔
G = A.copy()
gpA = [G]
for i in range(5):
    G = cv2.pyrDown(G)
    gpA.append(G)
 
G = B.copy()
gpB = [G]
for i in range(5):
    G = cv2.pyrDown(G)
    gpB.append(G)
 
# 产生Laplacian金字塔
lpA = [gpA[5]]
for i in range(5,0,-1):
    GE = cv2.pyrUp(gpA[i])
    L = cv2.subtract(gpA[i-1],GE)
    lpA.append(L)
 
lpB = [gpB[5]]
for i in range(5,0,-1):
    GE = cv2.pyrUp(gpB[i])
    L = cv2.subtract(gpB[i-1],GE)
    lpB.append(L)
 
# 合并
LS = []
for la,lb in zip(lpA,lpB):
    rows,cols,dpt = la.shape
    ls = np.hstack((la[:,0:cols//2], lb[:,cols//2:]))
    LS.append(ls)
 
# 重新构建图像
ls_ = LS[0]
for i in range(1,6):
    ls_ = cv2.pyrUp(ls_)
    ls_ = cv2.add(ls_, LS[i])
 
# 连接
real = np.hstack((A[:,:cols//2],B[:,cols//2:]))
 
cv2.imshow("LS",ls_)
cv2.imshow("Real",real)
 
cv2.waitKey()
cv2.destroyAllWindows()

图片原图如下：
分割线