利用泊松融合算法融合图片

最新推荐文章于 2022-08-29 10:35:21 发布

从此乐呵呵

最新推荐文章于 2022-08-29 10:35:21 发布

阅读量1.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：算法实现文章标签：融合图片

本文链接：https://blog.youkuaiyun.com/weixin_42578915/article/details/99685072

本文通过Python3演示如何使用泊松融合算法将图片融合，特别是将图二中的飞机抠取并融合到图一中。文章包含实现效果展示及详细步骤：创建相应文件夹结构，生成mask，拖动mask至目标图，最后执行泊松融合操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本文主要介绍实现方法，关于原理部分，请读者阅读：https://drustz.com/posts/2015/10/17/poisson/
采用python3来实现融合，举例说明。
图一
在这里插入图片描述
图二

本文主要做的事情是将图二的飞机抠出来融合到图一中，采用代码实现。
实现效果

实现代码

1.建立images等文件夹如下：
在这里插入图片描述
results：存放融合的结果图片。
source/data：存放如图二的图片。
source/mask：存放抠出来的如飞机的黑白mask图片。
target/data：存放如图一的图片。
target/mask：存放飞机的mask拖入到目标图的mask的合成图。

2.编写mask的生成代码

class MaskPainter():
    def __init__(self, image_dir, mask_dir):
        self.image_dir = image_dir
        self.mask_dir = mask_dir

    def _read_img(self, image_path):
        self.image = cv2.imread(image_path)
        self.image_copy = self.image.copy()

        self.mask = np.zeros(self.image.shape)
        self.mask_copy = self.mask.copy()
        self.size = 10
        self.to_draw = False

        self.window_name = "Draw mask. s:save; r:reset; q:quit"

    def _paint_mask_handler(self, event, x, y, flags, param):
        if event == cv2.EVENT_LBUTTONDOWN:
            self.to_draw = True

        elif event == cv2.EVENT_MOUSEMOVE:
            if self.to_draw:
                cv2.rectangle(self.image, (x - self.size, y - self.size),
                              (x + self.size, y + self.size),
                              (0, 255, 0), -1)