Изображения не сохраняются и не увеличиваются должным образом с помощью Keras - PullRequest
Новая
       42

Изображения не сохраняются и не увеличиваются должным образом с помощью Keras

0 голосов
/ 10 июля 2020

Я работаю над кодом классификации изображений и использовал ImageDataGenerator от Keras. У меня возникают проблемы с хранением изображений в пакетном режиме, и даже если я их решу (используя list вместо array), я все равно сталкиваюсь с проблемами, когда приходит часть datagen. Вот код для хранения изображений: 

for i in range(len(img_file)):
    file_name = list(img_file.iloc[i]['id'])
    idx = file_name[0]+'/'+file_name[1]+'/'+file_name[2]+'/'
    file = base_folder+idx+img_file.iloc[i]['id']+'.jpg'
    img = cv2.resize(cv2.imread(file),ip_shape)
    imgs[i,:,:,:] = img
    plt.imshow(imgs[i])
    plt.show()

Где imgs инициализируется как imgs = np.zeros((10, 300, 300, 3))

Когда я показываю изображения, я получаю следующее:

введите описание изображения здесь

Однако, когда я инициализирую imgs как list вместо array, сохраняю его в list, а затем конвертирую array, тогда он работает: 

imgs.append(img)
test_batch = np.asarray(imgs)
plt.imshow(test_batch[0])
plt.show()

Затем, когда я дохожу до стадии увеличения данных, то же самое происходит с обучающими изображениями. matplotlib не отображает изображение.

Я что-то упустил? Или мне просто продолжить обучение?

...