如何解决喀拉拉邦的U-net维数问题

Question

我试图用Unet分割视网膜血管，但是它有问题并抛出：

我使用了输入为（512,512）的U-net模型。我使用“ flow_from_directory”来配置图像的 target_size =（512,512）。

train_datagen = ImageDataGenerator(rescale = 1./255,
                                   shear_range = 0.2,
                                   zoom_range = 0.2,
                                   horizontal_flip = True)
test_datagen = ImageDataGenerator(rescale = 1./255)
train_set=train_datagen.flow_from_directory('C:/Users/User/Desktop/Thesis-Diabetic Retinopathy/Datasets/UNET-test -2/Train',
                                                 target_size=(512,512),
                                                 color_mode='grayscale',
                                                 batch_size=32,
                                                 class_mode='categorical',
                                                 shuffle=True)  

test_set = test_datagen.flow_from_directory('C:/Users/User/Desktop/Thesis-Diabetic Retinopathy/Datasets/UNET-test -2/Test',
                                            target_size = (512,512),
                                            batch_size = 32,
class_mode = 'categorical')


model=unet()
history=model.fit_generator(train_set,
                         samples_per_epoch = 8000,
                         nb_epoch = 25,
                         validation_data = test_set,
                         nb_val_samples = 2000)

但是我不明白里面的问题在哪里。模型是here并更改了input_size