Using Keras ImageDataGenerator class

[mavaylon1]

This doesn't seem to work when I use my own data with keras image data generator class. For context, I've tested this data with UNET and VGG and works. I also wonder if the output dimensions is correct. I'm using this for image segmentation and the images are (512,512,3) but the output_shape from this resnet model is (None,2).