画像を結合して分割

concat_and_split.py

import cv2
import pathlib

# 画像をnこずつ結合してm個に分割
n = 2
m = 3

infolder = pathlib.Path('INFOLDER')
inimgs = infolder.glob('*')
inimgs = sorted(inimgs, key=lambda x: x.stem)

for topidx in range(0, len(inimgs), n):
    # nとばしでインデックスを取る

    # nファイル読み込む
    image_list = []
    for fidx in range(0, n):

        img = cv2.imread(str(inimgs[fidx]))
        image_list.append(img)

    print(image_list)

    # 縦に結合の場合ｇ
    image_v = cv2.vconcat(image_list)
    # 横に結合の場合
    # image_h = cv2.hconcat(image_list)

    # サイズ取得
    h, w = image_v.shape[:2]

    # 縦に分割するときの画素数
    dx = int(h/m)

    # データをｍこに分割する。
    new_imgs = [image_v[dx * i:dx * (i+1), :] for i in range(m)]

    # できたやつを出力
    for count, new_img in enumerate(new_imgs):
        cv2.imwrite(f'OUTFOLDER/outfile_{topidx}_{topidx+n-1}_{count}.jpg', new_img)