首页 > 解决方案 > ImageDataGenerator中batch_size的作用

问题描述

我是 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)这意味着批次中只有一个图像。我无法理解。第二张图在哪里?如何显示批次的第二张图片。

标签: python-3.xmachine-learningkerasdeep-learningdata-augmentation

解决方案


推荐阅读