search

WAMAWAMA / TNSCUI2020-Seg-Rank1st

This is the source code of the 1st place solution for segmentation task in MICCAI 2020 TN-SCUI challenge.
481 stars 78 forks source link

使用自己数据集时报错 #34

Closed Llululu closed 2 years ago

Llululu commented 2 years ago

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

WAMAWAMA commented 2 years ago

原数据是单通道的，可能你的输入图像是RGB三通道的吧，那就把config对应的地方改成3即可（👉这里,训练代码的135行 img_ch 这个参数）

Llululu commented 2 years ago

哦哦好的我试试

Llululu commented 2 years ago

我的mask也是三通道的，是不是out_ch也要改成3呢

WAMAWAMA commented 2 years ago

我的mask也是三通道的，是不是out_ch也要改成3呢

没错，不过如果你的out是三通道的，可能这个代码原生不适合从头跑到尾（因为本来是针对单通道单目标分割任务），建议参考着来修改（其实这份代码拆开来用即可，没必要用全部，精髓就是smp这个库，别的操作都很常规）。

Llululu commented 2 years ago

嗯嗯，我还是转成一通道了