JianyingLi

Yeah, I think this code can not work for multi-class case. It looks like the shape of ground truth is (batch_size, n_row*n_col, num_class), but the shape of model output (conv_10)...

You can try more epochs to improve the acc . Another place to check is the function of adjustData. `def adjustData(img,mask,flag_multi_class,num_class): ` `#here# ` `elif(np.max(img) > 1): ` `img =...