Эффект batch_size в ImageDataGenerator - PullRequest
1 голос
/ 27 апреля 2020

Я новичок в Keras и пытаюсь увеличить объем данных, но застрял в самом начале.

У меня есть изображение, и я пытаюсь сделать его дополнения следующим образом:

datagen=image.ImageDataGenerator(rotation_range=20)
iter=datagen.flow(samples,batch_size=2)
batch=iter.next()
plt.imshow(batch[0].astype('uint8'))

Итак, я понимаю, что datagen является генератором, а iter является итератором для повторюсь, но мои сомнения касаются batch_size. Здесь the batch_size=2 означает, что во время каждой итерации создается партия из 2 изображений. Теперь я могу видеть первое изображение в пакете, используя batch[0], как показано выше, но не могу видеть второе изображение в пакете, используя batch[1]. Когда я проверяю batch.shape, отображается (1,399,640,3), что означает, что в пакете есть только одно изображение. Я не могу этого понять. Где второе изображение? Как я могу отобразить второе изображение партии.

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