ndarray преобразование объекта в массив, где объект содержит изображения переменного размера - PullRequest
0 голосов
/ 27 апреля 2018

Я хочу импортировать изображения разных размеров в массиве в формате (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.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...