mask数据集通道数不匹配
tan0427 opened this issue · 1 comments
tan0427 commented
作者你好，我跑你代码的时候，发现不管是训练还是测试模型的时候，mask_root都只能用你的那个irregular_mask数据集才不会报错，而当我用我自己的mask数据集时，就会出现
而当我把训练代码
image_in_channels=3改为9时，又会报出这个错误
@xiefan-guo 请问这个问题该怎么解决。
penguinXvch commented
把
mask = self.mask_files_transforms(mask)
改成
mask = self.mask_files_transforms(mask.convert('L'))