PythonでOpenCVとnumpyを使って画像の分割と結合

import sys, os, glob
import cv2
import numpy as np

# 分割の幅と高さ
split_width = 256
split_height = 256
# ディレクトリ
in_dir = "./input"
out_dir = "./output"
# コマンドライン引数
args = sys.argv
if (len(args)>=3):
    in_dir = args[1]
    out_dir = args[2]
# 出力ディレクトリの作成
if not(os.path.exists(out_dir)):
    os.mkdir(out_dir)
# ファイルの一覧
for f in glob.glob((in_dir+'/*.*')):
    base, ext = os.path.splitext(os.path.basename(f))
    im = cv2.imread(f)
    h, w = im.shape[:2]
    #print([h, w])
    w_n = w // split_width
    if ((w % split_width) > 0):
        w_n = w_n + 1
    h_n = h // split_height
    if ((h % split_height) > 0):
        h_n = h_n + 1
    #print([h_n, w_n])
    for y in range(h_n):
        for x in range(w_n):
            yy = y*split_height
            xx = x*split_width
            height = split_height
            if (yy+height) > h:
                height = h - yy
            width = split_width
            if (xx+width) > w:
                width = w - xx
            tmp = im[yy:yy+height, xx:xx+width]
            #print(tmp.shape[:2])
            out_file = os.path.join(out_dir, "{0}_split_{1:03}_{2:03}{3}".format(base, y, x, ext))
            print(out_file)
            cv2.imwrite(out_file, tmp)

import sys, os, glob
import cv2
import numpy as np
import re

# 分割の幅と高さ
split_width = 256
split_height = 256
# ディレクトリ
in_dir = "./output"
out_dir = "./result"
# コマンドライン引数
args = sys.argv
if (len(args)>=3):
    in_dir = args[1]
    out_dir = args[2]
# 出力ディレクトリの作成
if not(os.path.exists(out_dir)):
    os.mkdir(out_dir)
# ファイルの一覧
basename = ""
im_h = None
im_v = None
yy = -1
xx = -1
for f in glob.glob((in_dir+'/*.*')):
    base, ext = os.path.splitext(os.path.basename(f))
    print(f)
    r = re.findall(r'(.+?)_split_(\d+)_(\d+)', base)
    if (r):
        b = r[0][0]
        y = int(r[0][1])
        x = int(r[0][2])
        print([b, y, x])
        tmp = cv2.imread(f)
        if (yy != y and im_h is not None):
            if (im_v is None):
                im_v = im_h.copy()
            else:
                im_v = cv2.vconcat([im_v, im_h])
            print(im_v.shape)
            im_h = None
        if (basename == ""):
            basename = b+ext
        if (basename != (b+ext) and im_v is not None):
            out_file = os.path.join(out_dir, basename)
            cv2.imwrite(out_file, im_v)
            basename = (b+ext)
            im_n = None
            im_v = None
            yy = -1
            xx = -1
        if (im_h is None):
            im_h = tmp.copy()
        else:
            im_h = cv2.hconcat([im_h, tmp])
        yy = y
        xx = x
if (basename != "" and im_v is not None):
    out_file = os.path.join(out_dir, basename)
    if (im_v is not None and im_h is not None):
        im_v = cv2.vconcat([im_v, im_h])
    print(im_v.shape)
    cv2.imwrite(out_file, im_v)

import sys, os, glob
import sys, os, glob
import cv2
import numpy as np
import re

# 
# 異なるディレクトリの同名画像ファイルを左右に連結
# 

# 画像の幅と高さ
image_width = 256
image_height = 256

# ディレクトリ
in_dir1 = "./input1"
in_dir2 = "./input2"
out_dir = "./result"
# コマンドライン引数
args = sys.argv
if (len(args)>=4):
    in_dir1 = args[1]
    in_dir2 = args[2]
    out_dir = args[3]
# 出力ディレクトリの作成
if not(os.path.exists(out_dir)):
    os.mkdir(out_dir)
for f in glob.glob((in_dir1+'/*.*')):
    filename = os.path.basename(f)
    base, ext = os.path.splitext(filename)
    print(filename)
    f2 = os.path.join(in_dir2, filename)
    if (os.path.exists(f2) == False):
        print("{0} not found".format(f2))
        continue
    
    img1 = np.zeros((image_width,image_height), dtype=np.uint8)
    img2 = np.zeros((image_width,image_height), dtype=np.uint8)
    im1 = cv2.imread(f, cv2.IMREAD_GRAYSCALE)
    im2 = cv2.imread(f2, cv2.IMREAD_GRAYSCALE)
    
    h, w =im1.shape[:2]
    img1[0:h,0:w] = im1

    h, w =im2.shape[:2]
    img2[0:h,0:w] = im2
    result = cv2.hconcat([img1, img2])
    
    print(result.shape)
    out_file = os.path.join(out_dir, filename)
    cv2.imwrite(out_file, result)