如何使用Keras ImageDataGenerator获取X_rotated

时间:2018-04-24 19:00:12

标签: image rotation keras

嗨,我只想让MNIST的数据库随机旋转:

我的X是一个numpy数组(5000,1,28,28) 我希望X_rotated的顺序相同。

我已经做到了:

datagen = ImageDataGenerator(rotation_range = 360)

datagen.fit(X)

现在我如何得到我的X_rotated?

他们只解释如何使用纪元和批处理来制作棘手的东西我只想回到我的阵列,其中每个图像随机旋转,没有什么棘手的。 我不明白为什么在教程上他们只解释如何制作棘手的东西而不是基础...

https://keras.io/preprocessing/image/

1 个答案:

答案 0 :(得分:0)

NumpyArrayIteradorDirectoryIterator对象与任何python迭代器非常相似:

g = ImageDataGenerator(...)
d = g.flow(..., batch_size=256, shuffle=False)

# flow all batches through the iterator,
# then zip all inputs and outputs, respectively.
batches = zip(*(next(d) for _ in range(len(d))))

# concatenate all inputs and outputs, respectively.
x, y = (np.concatenate(b) for b in batches)

print(x.shape, y.shape)

这应输出类似于此的内容:

  

(5000,1,28,28)(5000,?)