opencv可视化学习笔记(一):绘制多图在一个画布上

介绍

这个代码可直接将三张图片绘制在同一画布，只需要改动ROOT和图片路径即可。

具体代码

# coding=utf-8
from PIL import Image, ImageDraw, ImageFont
import cv2
import numpy as np


def jigsaw(imgs, direction="horizontal", gap=0):
    imgs = [Image.fromarray(img) for img in imgs]
    w, h = imgs[0].size
    if direction == "horizontal":
        result = Image.new(imgs[0].mode, ((w+gap)*len(imgs)-gap, h))
        for i, img in enumerate(imgs):
            result.paste(img, box=((w+gap)*i, 0))
    elif direction == "vertical":
        result = Image.new(imgs[0].mode, (w, (h+gap)*len(imgs)-gap))
        for i, img in enumerate(imgs):
            result.paste(img, box=(0, (h+gap)*i))
    else:
        raise ValueError("The direction parameter has only two options: horizontal and vertical")
    return np.array(result)


if __name__ == '__main__':
    ROOT = 'F:\python\HW_Match\LPDNet/'
    img1 = cv2.imread(ROOT+"data1/ori.png")
    img2 = cv2.imread(ROOT+"data1/pre1.png")
    img3 = cv2.imread(ROOT+"data1/pre2.png")
    img = jigsaw([img1, img2, img3])
    cv2.imwrite("test.png", img)

结果展示