from typing import Tuple
import cv2
import numpy as np
import argparse
import os
parser = argparse.ArgumentParser('将目录中的图片生成固定大小的图片阵列')
parser.add_argument('inputDir', type=str,
help='the directory stores the pictures!')
parser.add_argument('format', type=str, help='formet of picture to process!')
parser.add_argument('row', type=int, help='the row num of picture array!')
parser.add_argument('col', type=int, help='the column num of picture array!')
parser.add_argument('outPath', type=str,
help='the path to store the picture array!')
parser.add_argument('--label', type=bool,
help='the flag indicate whether note the picture with its name!', default=True)
args = parser.parse_args()
inputDir = args.inputDir
format = args.format
row = args.row
col = args.col
outPath = args.outPath
imgfiles = os.listdir(inputDir)
imgfiles = list(filter(lambda imgfile: imgfile.endswith(format), imgfiles))
if row*col != len(imgfiles):
print('The row num and row num don\'t match the num of images!')
exit()
imgArray = list()
index = 0
for i in range(row):
rowArray = list()
for j in range(col):
filename = imgfiles[index]
imgfile = os.path.join(inputDir, filename)
img = cv2.imread(imgfile)
filename = filename[0:filename.rindex('.')]
img = cv2.putText(img=img, text=filename, org=(10, 30), fontFace=cv2.FONT_HERSHEY_SIMPLEX,
fontScale=1, color=(0, 0, 0), thickness=2, lineType=cv2.LINE_AA, bottomLeftOrigin=False)
rowArray.append(img)
index += 1
rowArray = np.concatenate(rowArray, axis=1)
imgArray.append(rowArray)
imgArray = np.concatenate(imgArray, axis=0)
print(imgArray.shape)
cv2.imwrite(outPath, imgArray)
print('finished!')
调用方式
python imgArray.py ./ 'tif' 5 4 demo.tif
效果