python-cv 多张图片写到一张图片

python-cv 多张图片写到一张图片

#!/usr/bin/python3.6.10
# -*- coding: utf-8 -*-
# @Author  : C
# @Time    : 2022/1/7 10:05
# @File    : 多张图片写到一张图片.py

import cv2
from pylab import *

# img1 = cv2.imread('lena.jpg',cv2.IMREAD_COLOR)
# img2 = cv2.imread('lena.jpg',cv2.IMREAD_GRAYSCALE)
# img3 = cv2.imread('lena.jpg',cv2.IMREAD_UNCHANGED)
# img4 = cv2.imread('lena.jpg')

# res = np.zeros((1920,1080,3),dtype=np.float)
res = np.full((540,960,3), 114, dtype=np.uint8)
print(res)
img5 = cv2.imread('./images/000171.jpg')
img5 = cv2.resize(img5,(960,540))
print(img5.shape)
img6 = cv2.imread('./images/000191.jpg')
img6 = cv2.resize(img6,(960,540))

img7 = cv2.imread('./images/000196.jpg')
img7 = cv2.resize(img7,(960,540))

img8 = cv2.imread('./images/000226.jpg')
img8 = cv2.resize(img8,(960,540))
h1 = [img5, img6, img7, img8]

htitch1= np.vstack(h1[:2])
htitch2= np.vstack(h1[2:])

vtitch = np.hstack((htitch1, htitch2))
# vtitch = vtitch.transpose((1,0,2))
cv2.imshow("test1", vtitch)
cv2.imwrite('111.jpg',vtitch)

# cv2.imshow("test2",img5)
cv2.waitKey(0)
cv2.destroyAllWindows()