python批量读取图片数据集_从图像数据集读取图片并拼接成大图

【实例简介】

从图像数据集读取图片并拼接成大图

【实例截图】

【核心代码】

import PIL.Image as Image

import os

IMAGES_PATH = 'D:/17flowers_data/17flowers/'  # 图片集地址

IMAGES_FORMAT = ['.jpg']  # 图片格式

IMAGE_SIZE = 256  # 每张小图片的大小

IMAGE_ROW = 10 # 图片间隔，也就是合并成一张图后，一共有几行

IMAGE_COLUMN =10  # 图片间隔，也就是合并成一张图后，一共有几列

IMAGE_SAVE_PATH = 'final.jpg'  # 图片转换后的地址

# 获取图片集地址下的所有图片名称

image_names = [name for name in os.listdir(IMAGES_PATH) for item in IMAGES_FORMAT if

os.path.splitext(name)[1] == item]

# 定义图像拼接函数

def image_compose():

to_image = Image.new('RGB', (IMAGE_COLUMN * IMAGE_SIZE, IMAGE_ROW * IMAGE_SIZE)) #创建一个新图

# 循环遍历，把每张图片按顺序粘贴到对应位置上

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)

to_image.paste(from_image, ((x - 1) * IMAGE_SIZE, (y - 1) * IMAGE_SIZE))

return to_image.save('D:\Flower.jpg') # 保存新图

image_compose() #调用函数