Я хочу импортировать изображения разных размеров в массиве в формате (Number of images, row_size, col_size)
. Я использую следующий код, чтобы сделать это.
import os,cv2, numpy as np
PATH = os.getcwd()
data_path = PATH+'/data'
data_dir_list = os.listdir(data_path)
img_data_list=[]
for dataset in data_dir_list:
img_list=os.listdir(data_path+'/'+ dataset)
print ('Loaded the images of dataset-'+'{}\n'.format(dataset))
for img in img_list:
input_img=cv2.imread(data_path + '/'+ dataset + '/'+ img,0)
img_data_list.append(input_img)
img_data = np.array(img_data_list)
Когда я работаю с изображениями одинакового размера, я получаю правильный результат в img_data
, как в примере size=(1000,50,50)
и type=uint8
, но когда я работаю с изображениями разных размеров (все 1000 изображений имеют разные размеры, такие как 34 * 45, 25 * 43 ........) я получил img_data
как size=(1000,)
type=object
value=ndarray object of numpy module
. Я работаю над глубоким обучением в keras и python.