图像金字塔图层叠加实现图像融合（基于opencv-python）

最新推荐文章于 2024-08-15 08:00:00 发布

是小陈啊阿啊

最新推荐文章于 2024-08-15 08:00:00 发布

阅读量581

点赞数 1

文章标签： python

本文链接：https://blog.csdn.net/m0_58409035/article/details/129452638

版权

看了很多代码都是将图像左右相加，而不是图层直接相加，其实改几行代码就可以，用CV2.add函数就可以，同时取col和row的时候取全图就行。

'''

laplace金字塔图像融合技术
'''
import cv2
import numpy as np, sys

A = cv2.imread('DCP.jpg')
B = cv2.imread('DCP.jpg')
# generate Gaussian pyramid for A
G = A.copy()
gpA = [G]
for i in range(6):
    G = cv2.pyrDown(G)
    gpA.append(G)
# generate Gaussian pyramid for B
G = B.copy()
gpB = [G]
for i in range(6):
    G = cv2.pyrDown(G)
    gpB.append(G)
# generate Laplacian Pyramid for A
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)
# generate Laplacian Pyramid for B
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)
# Now add left and right halves of images in each level
# numpy.hstack(tup)
# Take a sequence of arrays and stack them horizontally
# to make a single array.
LS = []
for la, lb in zip(lpA, lpB):
    rows, cols, dpt = la.shape

    # ls = np.hstack((la[:, :cols // 2], lb[:, cols // 2:]))
    ls = np.hstack((la, lb))

    # ls = np.hstack((la[:, :cols // 4], lb[:, cols // 4: cols//2], la[:, cols//2:3 * cols//4], lb[:, 3*cols//4: ]))
    LS.append(ls)
# now reconstruct
ls_ = LS[0]
for i in range(1, 6):
    ls_ = cv2.pyrUp(ls_)
    ls_ = cv2.add(ls_, LS[i])
# image with direct connecting each half
# real = np.hstack((A[:, :cols // 2], B[:, cols // 2:]))
real=cv2.add(A, B)
# real = np.hstack((A, B))
# real = np.hstack((A[:,:cols//4],B[:,cols//4: cols//2], A[:, cols//2: 3*cols//4], B[:, 3*cols//4: ]))
cv2.imshow('inny_nero.jpg', ls_)
cv2.imshow('ini_chaochao.jpg', real)

cv2.waitKey(0)
cv2.destroyAllWindows()
cv2.imwrite('real.jpg', real)