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)