数据处理---图像拼接（也可处理后再拼接）

# 数据拼接
import PIL.Image as Image
from PIL import Image, ImageDraw, ImageFont
import os

# 定义图像拼接函数
def image_compose(i):
    to_image = Image.new('RGB', (IMAGE_COLUMN * image_heiht, IMAGE_ROW * image_width))  # 创建一个新图
    # 循环遍历，把每张图片按顺序粘贴到对应位置上
    for y in range(1, IMAGE_ROW + 1):
        for x in range(1, IMAGE_COLUMN + 1):
            from_image = Image.open(IMAGES_PATH + "/"+image_names[IMAGE_COLUMN * (y - 1) + x - 1])#.resize((IMAGE_SIZE, IMAGE_SIZE), Image.ANTIALIAS)
            draw = ImageDraw.Draw(from_image)
            typeface = ImageFont.truetype('STKAITI.TTF', 40)  # 参数： 字体  字体大小  STXINGKA.TTF
            draw.text((20,550), image_names[IMAGE_COLUMN * (y - 1) + x - 1] , fill=(255, 0, 0), font=typeface)
            to_image.paste(from_image, ((x - 1) * image_heiht, (y - 1) * image_width))
    return to_image.save("D:\PycharmProjects\Classification1\datasets/train\风信子/"+str(i)+'.jpg')  # 保存新图
for i in range(100):
    IMAGES_FORMAT = ['.jpg', '.JPG']  # 图片格式
    IMAGE_SIZE = 128  # 每张小图片的大小
    IMAGE_ROW = 3  # 图片间隔，也就是合并成一张图后，一共有几行
    IMAGE_COLUMN = 2  # 图片间隔，也就是合并成一张图后，一共有几列
    IMAGES_PATH = 'D:\PycharmProjects\Classification1\datasets/train2/hyacinth/' + str(i) + ''
    image_names = [name for name in os.listdir(IMAGES_PATH) for item in IMAGES_FORMAT if
                   os.path.splitext(name)[1] == item]
    print(image_names)
    # 获取图片集地址下的所有图片名称
    # 简单的对于参数的设定和实际图片集的大小进行数量判断
    if len(image_names) != IMAGE_ROW * IMAGE_COLUMN:
        raise ValueError("合成图片的参数和要求的数量不能匹配！")
    image_heiht, image_width = Image.open(IMAGES_PATH + "/" + image_names[0]).size
    image_compose(i)