Python脚本25：将两张图片拼在一起

import PIL.Image as Image
import os
# from IPython import embed
# embed()

# 定义图像拼接函数
def image_compose(imag, imag_1):
    src = os.path.join(os.path.abspath(IMAGE_SAVE_PATH), img)
    to_image = Image.new('RGB', (2 * h, 1 * w))  # 创建一个新图
    # 把两张图片按顺序粘贴到对应位置上
    rom_image = Image.open(IMAGES_PATH + imag).resize((h, w), Image.ANTIALIAS)
    rom_image_1 = Image.open(IMAGES_PATH_1 + imag_1).resize((h, w), Image.ANTIALIAS)
    to_image.paste(rom_image, (0, 0))
    to_image.paste(rom_image_1, (h, 0))
    to_image.save(src)  # 保存新图，还是原来的名称

if __name__ == '__main__':
    IMAGES_PATH = '/home/xl/workstation/Datasets/cuhk03/images_labeled/'  # 原图片集地址
    IMAGES_PATH_1 = '/home/xl/workstation/Datasets/cuhk03/low_images_labeled/'  # 低光图片集地址
    IMAGE_SAVE_PATH = '/home/xl/workstation/Datasets/cuhk03/join_images_labeled/'  # 保存新图片地址
    IMAGES_FORMAT = ['.png', '.PNG']  # 图片格式
    list_n = []
    w = 128  # 每张小图片的大小
    h = 64
    # 获取图片集地址下的所有图片名称
    image_names = [name for name in os.listdir(IMAGES_PATH) for item in IMAGES_FORMAT if
                   os.path.splitext(name)[1] == item]

    image_names_1 = [name for name in os.listdir(IMAGES_PATH_1) for item in IMAGES_FORMAT if
                     os.path.splitext(name)[1] == item]

    # 文件名相同就调用拼接函数
    for img in image_names:
        for img_1 in image_names_1:
            if img == img_1:
                image_compose(img, img_1)