Создание массива ранга 4 numpy из плоского списка пикселей для использования в ImageDataGenerator в Keras, а затем в matplotlib

Question

У меня есть фрейм данных, состоящий из серых (c = 1) изображений размером 137*236, что в сумме составляет 32332 пикселей. Я хочу увеличить каждое изображение с помощью ImageDataGenerator(), но при использовании кода ниже

samples = np.expand_dims(df.iloc[[65],1:].values.reshape(137,236,1), 0)
# create image data augmentation generator
datagen = ImageDataGenerator(width_shift_range=[-200,200])
# prepare iterator
it = datagen.flow(samples, batch_size=1)
# generate samples and plot
for i in range(9):
    # define subplot
    pyplot.subplot(330 + 1 + i)
    # generate batch of images
    batch = it.next()
    # convert to unsigned integers for viewing
    image = batch[0].astype('uint8')
    print(image.shape)
    pyplot.imshow(image)
    # show the figure
    pyplot.show()

появляется ошибка как Invalid dimensions for image data, а при использовании samples = np.expand_dims(df.iloc[[65],1:].values.reshape(137,236), 0)? 1010 отображаются ошибки как 'Input data in `NumpyArrayIterator` should have rank 4. You passed an array with shape', (1, 137, 236) *

Что здесь не так?