Как получить мой X_rotated с Keras ImageDataGenerator

Question

Привет, я просто хочу, чтобы моя база данных MNIST была повернута случайным образом:

У меня есть X, который является массивом пустышек (5000, 1, 28, 28). Я хочу, чтобы X_rotated был в том же порядке и в том же измерении.

Я сделал это:

datagen = ImageDataGenerator (вращение_диапазона = 360)

datagen.fit (X)

Теперь, как мне получить X_rotated?

Они только объясняют, как делать сложные вещи с эпохами и партиями. Я просто хочу вернуть свой массив, где каждое изображение случайно вращало больше ничего, ничего сложного.Я не понимаю, почему в уроках они только объясняют, как делать сложные вещи, но не основы ...

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

Answer 1

Объекты NumpyArrayIterador и DirectoryIterator очень похожи на любой итератор 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,?)