mqray

I have found some useful solution, you can refer `https://github.com/wuyang0329/unet`

I found you did some modification in data.py, including: ``` def adjustData(img,mask,flag_multi_class,num_class): if(flag_multi_class): #print(type(img)) img = img / 255. mask = mask[:,:,:,0] if(len(mask.shape) == 4) else mask[:,:,0] mask[(mask!=0.)&(mask!=255.)&(mask!=128.)] = 0....

What's more, I guess that the function called labelVisualize in data.py may wrong. ``` def labelVisualize(num_class, color_dict, img): img_out = np.zeros(img[:,:,0].shape + (3,)) for i in range(img.shape[0]): for j in...

Besides, when we use flow_from_directory, the class mode was ser as grayscale. Notice that in your dataset, the photos' mode is rgb, I guess maybe the color_mode should change. In...