Keras custom generator: Error when checking input: expected conv2d_1_input to have 4 dimensions, but got array with shape (256, 1)

Question

I have a large dataset (30 000 images, 80kB each), and I\'m trying to load it into my model in batches in order to not run of out memory on my GPU. My single image is a