python遍历文件夹下所有图片_Python--把文件夹下的图片全部拼接成一张大图

输入：

1、几行几列

2、图片路径 （不遍历子文件夹）

import os

import cv2

import numpy as np

import PIL.Image as Image

import matplotlib.pyplot as plt

img_type = ['.jpg','.JPG','.png','.PNG','.bmp','.BMP']#可继续添加图片类型

#输入几行几列

ROW = 4

COL = 5

def resize_blank_img(srcimg, dstpath):

img = cv2.imread(srcimg)

blankimg = np.zeros(np.shape(img), np.uint8)

blankimg[:, :, 0] = 255

blankimg[:, :, 1] = 255

blankimg[:, :, 2] = 255

num = [os.path.join(dstpath,imgpath) for imgpath in os.listdir(dstpath) if os.path.splitext(imgpath)[1] in img_type]

cv2.imwrite(dstpath + "\\" + str(len(num)+1) + ".jpg", blankimg)

def image_compose(image_path):

if not os.path.isdir(image_path):

return -1

imgpath_vec = [os.path.join(image_path,imgpath) for imgpath in os.listdir(image_path) if os.path.splitext(imgpath)[1] in img_type]

#1、使用平均的width，heigth或者可以自定义width，heigth

avg_width = 0

avg_heigth = 0

if avg_width == 0 or avg_heigth == 0:

size = []

for item in imgpath_vec:

size.append((Image.open(item)).size)

sum_width = sum_heigth = 0

for item in size:

sum_width += item[0]

sum_heigth += item[1]

avg_width = int(sum_width/(len(size)))

avg_heigth = int(sum_heigth/(len(size)))

avg_size = (avg_width,avg_heigth)

#2、resize图片大小

vec = [os.path.join(image_path, imgpath) for imgpath in os.listdir(image_path) if

os.path.splitext(imgpath)[1] in img_type]

while (len(vec)) < COL * ROW:

vec = [os.path.join(image_path, imgpath) for imgpath in os.listdir(image_path) if

os.path.splitext(imgpath)[1] in img_type]

resize_blank_img(vec[0],image_path)

imgs = []

for item in vec:

imgs.append((Image.open(item)).resize(avg_size,Image.BILINEAR))

#3、拼接成大图

result_img = Image.new(imgs[0].mode,(avg_width * COL,avg_heigth * ROW))

index = 0

for i in range(COL):

for j in range(ROW):

result_img.paste(imgs[index],(i * avg_width, j * avg_heigth))

index+=1

#4、显示拼接结果

plt.imshow(result_img)

plt.show()

# result_img.save(path)#保存结果

if __name__ == "__main__":

path = r'E:\image_processing_data\png'#输入你的图片路径

image_compose(path)