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）

赶时间先不实现了，有没有更好的办法以后做图像更深入了可能会想到。