Using Keras ImageDataGenerator class
mavaylon1 opened this issue · 0 comments
mavaylon1 commented
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).