python 将多张小图合并拼接成一张大图

import os
import cv2
import glob
import copy
import numpy as np

prefix = 're00'
imagename = 'restored.jpg'
rootdir = './ourModelResult'
savedir = './restored'

for i in range(32):
    dirprefix = ''.join([prefix, f'{i:02d}-*'])
    readpath = os.path.join(rootdir, dirprefix)
    dirlist = glob.glob(readpath)

    j = 0
    step = 8
    blockimage = None
    fullimage = None
    for dir in dirlist:
        imagepath = os.path.join(dir, imagename)
        print(imagepath)
        image = cv2.imread(imagepath)

        if blockimage is None:
            blockimage = image
        else:
            blockimage = np.concatenate([blockimage, image], axis=1)

        j = j + 1
        if j % step == 0:
            if fullimage is None:
                fullimage = copy.deepcopy(blockimage)
                print(fullimage.shape)
            else:
                fullimage = np.concatenate([fullimage,blockimage], axis=0)
                print(fullimage.shape)

            blockimage = None

    savepath = os.path.join(savedir, str(i+1)+'.jpg')
    cv2.imwrite(savepath, fullimage)