ValueError: expected cropping2d_1 to have shape (360, 480, 2) but got array with shape (360, 480, 1)

我发现一系列这种keras的 Error when checking target 的bug有百分之90来源于onehot编码的问题！
谨此纪念调了两天的bug，希望用这个经历警醒自己，没有调不通的bug，只有查资料查的不够多的，不细心看bug的人！！
我做的是图像分割的问题

更改前代码

def adjustData(original,mask):
    original = original/255
    mask = mask/255
    #mask = to_categorical(mask, 2)
    mask[mask > 0.5] = 1
    mask[mask < 0.5] =0
    return(original,mask)

此时label是0或1，但我们需要One hot
改成

def adjustData(original,mask):
    original = original/255
    mask = mask/255
    mask = to_categorical(mask, 2)
    #mask[mask > 0.5] = 1
    #mask[mask < 0.5] =0
    return(original,mask)

弄多分类的小伙伴把to_categorical的参数调一下噢

感觉有帮助的小伙伴点个赞再走嗷~

最后感谢博主提供思路 万级阅读量的一篇blog

最后希望人间再无bug嘿嘿