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