1. 程式人生 > 程式設計 >numpy陣列做圖片拼接的實現(concatenate、vstack、hstack)

numpy陣列做圖片拼接的實現(concatenate、vstack、hstack)

兩種方法拼接

#img = np.vstack((img,img2))  # vstack按垂直方向,hstack按水平方向
img = np.concatenate((img,img2),axis=0)  # axis=0 按垂直方向,axis=1 按水平方向

統一圖片大小,保證陣列維度一致避免拼接失敗。 把圖片全部調整成第一張圖的寬高

def img_size(image_names,width,height):
  for i in image_names:
    img = cv2.imread(os.path.join(img_path,i))
    img_resize = cv2.resize(img,(width,height),interpolation=cv2.INTER_CUBIC)
    cv2.imwrite(os.path.join(img_path,i),img_resize)
    print(os.path.join(img_path,i))

完整案例,拼接資料夾中的所有圖片

import cv2
import os
import numpy as np

def img_size(image_names,i))

if __name__ == '__main__':
  img_path = r'F:\studytest'
  image_names = [name for name in os.listdir(img_path) if os.path.splitext(name)[1] == ".jpg"]
  img1 = cv2.imread(os.path.join(img_path,image_names[0]))
  width,height = img1.shape[:2][::-1]
  img_size(image_names,height)
  img = img1

  for i in range(1,len(image_names)):
    img_page = image_names[i]
    img2 = cv2.imread(os.path.join(img_path,img_page))
    #img = np.vstack((img,img2))  # vstack按垂直方向,hstack按水平方向
    img = np.concatenate((img,axis=0)  # axis=0 按垂直方向,axis=1 按水平方向
  cv2.imwrite(os.path.join(img_path,"res.jpg"),img)
  # cv2.imshow("img",img)
  # cv2.waitKey()
``

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。