Closed Llululu closed 2 years ago
原数据是单通道的,可能你的输入图像是RGB三通道的吧,那就把config对应的地方改成3即可(👉这里,训练代码的135行 img_ch
这个参数)
哦哦好的我试试
我的mask也是三通道的,是不是out_ch也要改成3呢
我的mask也是三通道的,是不是out_ch也要改成3呢
没错,不过如果你的out是三通道的,可能这个代码原生不适合从头跑到尾(因为本来是针对单通道单目标分割任务),建议参考着来修改(其实这份代码拆开来用即可,没必要用全部,精髓就是smp这个库,别的操作都很常规)。
嗯嗯,我还是转成一通道了
RuntimeError: Given groups=1, weight of size [56, 1, 3, 3], expected input[16, 3, 257, 257] to have 1 channels, but got 3 channels instead