history = model.fit_generator(
aug.flow(x_train, y_train, batch_size=BS),
validation_data=(x_test, y_test),
steps_per_epoch=len(x_train) // BS,
epochs=EPOCHS, verbose=1
)
VaueError: ('Input data in `NumpyArrayIterator` should have rank 4. You passed an array with shape', (2361, 0))