Python 图片简单拼接

import numpy as np
import cv2 as cv
import os
import argparse

parser = argparse.ArgumentParser(description='merge picture')
parser.add_argument("-i","--input", type=str, default="")
parser.add_argument("-o","--output", type=str, default="default")
args = parser.parse_args()

#读取仅存放需要操作的图片的file_name目录下的文件名
#通过cv.imread读取图像数据并存放在列表pic中
#最后利用np.vstack垂直排列所有图像数据（数组）

pic = []
def file_name(file_dir):
    for root, dirs, files in os.walk(file_dir):
        n = len(files)
        for i in files:
            pic.append(cv.imread(i))

file_name(args.input)

merge_pic = np.vstack(pic) #np.vstack换成np.hstack可以将上下拼接改成左右拼接 
cv.imwrite(args.output,merge_pic)