python 圖像拼接

import cv2
import numpy as np
import pandas as pd
from PIL import Image

img1 = cv2.imread('lena.bmp', cv2.IMREAD_GRAYSCALE)
cv2.namedWindow("lena", cv2.WINDOW_AUTOSIZE)
cv2.imshow("lena", img1)
print(img1.shape)

img2 = cv2.imread('Cameraman.bmp', cv2.IMREAD_GRAYSCALE)
cv2.namedWindow("Cameraman", cv2.WINDOW_AUTOSIZE)
cv2.imshow("Cameraman", img2)
print(img2.shape)

img3 = cv2.imread('house.tif', cv2.IMREAD_GRAYSCALE)
cv2.namedWindow("house", cv2.WINDOW_AUTOSIZE)
cv2.imshow("house", img3)
print(img3.shape)

def concat_pic(img):
    # numpy連接 直接用numpy的ndarray數組合並方法concatenate()
    img_01 = np.vstack(img)  # 縱向連接
    img_02 = np.concatenate(img, axis=1)  # 1 橫向連接  0 縱向連接

    # pandas連接 先將圖片由ndarray轉換成dataframe格式，使用pandas的數據集連接方法concat()處理后，再由dataframe格式轉回ndarray格式
    # ndarray to dataframe
    df = []
    for i in range(len(img)):
        df.append(pd.DataFrame(img[i]))
    dff = pd.concat(df, axis=1)  # 1 橫向連接  0 縱向連接
    # dataframe to ndarray
    img_03 = np.array(dff)
    print(img_03.shape)

    # 用opencv打開的圖片是numpy.ndarray格式
    cv2.namedWindow("hecheng", cv2.WINDOW_AUTOSIZE)
    cv2.imshow("hecheng", img_03)

    keyx = cv2.waitKey(0)
    print(keyx)
    if keyx == 27:
        print("你按下了Esc鍵")
        cv2.destroyAllWindows()
    elif keyx == ord('s'):
        print("你保存了這張圖像")
        cv2.imwrite("origin.png", img_03)

if __name__ == '__main__':
    img = [img1, img2, img3]
    concat_pic(img)

 

 

這只是用numpy 或 pandas 拼接了行或列，意思是只能橫着接或者豎着接，如果想要m*n張圖片該怎么拼接呢？

暫時的想法只是 比如說3*2一共六張圖片，先三張 三張地橫着拼成了兩大張（2張3*1），然后兩大張圖片再豎着接成一張完整的圖（3*2）

趕時間先不實現了，有沒有更好的辦法以后做圖像更深入了可能會想到。