Python 输出多福排列图像

# -*- coding: utf-8 -*-
import numpy as np
from PIL import Image
import cv2
from os.path import dirname as ospdn
import os
import os.path as osp

def may_make_dir(path):
  """
  Args:
    path: a dir, or result of `osp.dirname(osp.abspath(file_path))`
  Note:
    `osp.exists('')` returns `False`, while `osp.exists('.')` returns `True`!
  """
  # This clause has mistakes:
  # if path is None or '':

  if path in [None, '']:
    return
  if not osp.exists(path):
    os.makedirs(path)
    
def read_im(im_path):
  # shape [H, W, 3]
  im = np.asarray(Image.open(im_path.rstrip()))
  # Resize to (im_h, im_w) = (128, 64)
  resize_h_w = (120, 160)
  if (im.shape[0], im.shape[1]) != resize_h_w:
    im = cv2.resize(im, resize_h_w[::-1], interpolation=cv2.INTER_LINEAR)
  # shape [3, H, W]
  im = im.transpose(2, 0, 1)
  if (im.shape[0] is not 3):
      im = im[0:3,:,:]
  return im

def save_im(im, save_path):
  """im: shape [3, H, W]"""
  may_make_dir(ospdn(save_path))
  im = im.transpose(1, 2, 0)
  Image.fromarray(im).save(save_path)
  
  
def make_im_grid(ims, n_rows, n_cols, space, pad_val):
  """Make a grid of images with space in between.
  Args:
    ims: a list of [3, im_h, im_w] images
    n_rows: num of rows
    n_cols: num of columns
    space: the num of pixels between two images
    pad_val: scalar, or numpy array with shape [3]; the color of the space
  Returns:
    ret_im: a numpy array with shape [3, H, W]
  """
  assert (ims[0].ndim == 3) and (ims[0].shape[0] == 3)
  assert len(ims) <= n_rows * n_cols
  h, w = ims[0].shape[1:]
  H = h * n_rows + space * (n_rows - 1)
  W = w * n_cols + space * (n_cols - 1)
  if isinstance(pad_val, np.ndarray):
    # reshape to [3, 1, 1]
    pad_val = pad_val.flatten()[:, np.newaxis, np.newaxis]
  ret_im = (np.ones([3, H, W]) * pad_val).astype(ims[0].dtype)
  for n, im in enumerate(ims):
    r = n // n_cols
    c = n % n_cols
    h1 = r * (h + space)
    h2 = r * (h + space) + h
    w1 = c * (w + space)
    w2 = c * (w + space) + w
    ret_im[:, h1:h2, w1:w2] = im
  return ret_im

images_dir = 'D:/images'
save_path = 'D:/results.png'
n_rows = 10
n_cols = 10

dirs_files = os.listdir(images_dir)
dirs_files.sort()
image_names = []
for dir_file in dirs_files:
    image_path = os.path.join(images_dir, dir_file)
    if image_path.endswith('.jpg'):
        image_names.append(image_path)
q_im_paths = image_names[0:n_rows*n_cols]

ims = []
for q_im_path in q_im_paths:
    im = read_im(q_im_path)
    ims.append(im)
im = make_im_grid(ims, n_rows, n_cols, 8, 255)
save_im(im, save_path.rstrip())