Read multiple image data to ndarray - (n_samples,height,width,channels)

read_multi_image_data.py

import os
import numpy as np
import glob
import cv2

image_dir = './images/'
search_pattern = '*.png'
datas = []
for image_path in glob.glob(os.path.join(image_dir,search_pattern)):
    # (height,width,channels)
    data = cv2.imread(image_path)
    # (1,height,width,channels)
    data_expanded = np.expand_dims(data,axis=0)
    datas.append(data_expanded)
# (n_samples,height,width,channels)
image_datas = np.concatenate(datas,axis=0)