fit_generator标志停止迭代问题，并且预测图像为空

Question

我正在尝试为医学图像运行unet架构以进行细分。我使用了https://github.com/zhixuhao/unet中的代码。我的图片的实际尺寸为800x600，我已经将图片的尺寸调整为800x400以便进行培训。当我尝试测试8张图像时，预测停止迭代会出现错误。它成功运行了1张图像，但预测的输出图像为空。有人帮我解决了这个问题。

代码取自https://github.com/zhixuhao/unet 在data.py中完成的修改：

def trainGenerator(batch_size,train_path,image_folder,mask_folder,aug_dict,image_color_mode = "grayscale",mask_color_mode = "grayscale",image_save_prefix = "image",mask_save_prefix = "mask",flag_multi_class = False,num_class = 2,save_to_dir = None,target_size = (400,800),seed = 1):

def testGenerator(test_path,num_image = 9, target_size = (400,800), flag_multi_class = False,as_gray = True):

图像尺寸从（256,256）更改为（400,800）

在model.py->

中完成的修改

def unet(pretrained_weights = None,input_size = (400,800,1)):