写出普拉斯字塔融合算法python代码

普拉斯字塔融合算法（PyrBlend）是一种图像融合算法，用于将两幅图像进行无缝融合。下面是一个简单的Python代码示例，演示了如何使用OpenCV库实现普拉斯字塔融合算法：

import cv2
import numpy as np

def pyr_blend(img1, img2, levels=6):
    # 生成高斯金字塔
    gp1 = [img1]
    gp2 = [img2]
    for i in range(levels):
        img1 = cv2.pyrDown(img1)
        img2 = cv2.pyrDown(img2)
        gp1.append(img1)
        gp2.append(img2)

    # 生成拉普拉斯金字塔
    lp1 = [gp1[levels-1]]
    lp2 = [gp2[levels-1]]
    for i in range(levels-1, 0, -1):
        img1 = cv2.pyrUp(gp1[i])
        img2 = cv2.pyrUp(gp2[i])
        lap1 = cv2.subtract(gp1[i-1], img1)
        lap2 = cv2.subtract(gp2[i-1], img2)
        lp1.append(lap1)
        lp2.append(lap2)

    # 融合拉普拉斯金字塔
    blended = []
    for l1, l2 in zip(lp1, lp2):
        rows, cols, _ = l1.shape
        blended.append(np.hstack((l1[:, :cols//2], l2[:, cols//2:])))

    # 重建图像
    result = blended[0]
    for i in range(1, levels):
        result = cv2.pyrUp(result)
        result = cv2.add(result, blended[i])

    return result

# 读取两幅图像
img1 = cv2.imread('image1.jpg')
img2 = cv2.imread('image2.jpg')

# 调用普拉斯字塔融合算法
result = pyr_blend(img1, img2)

# 显示结果
cv2.imshow('PyrBlend Result', result)
cv2.waitKey(0)
cv2.destroyAllWindows()

请注意，这只是一个简单的示例代码，实际应用中可能需要根据具体需求进行调整和优化。另外，你需要安装OpenCV库才能运行以上代码。如果你还有其他问题，请继续提问。