RuntimeError: Given groups=1, weight of size 64 7 7 7, expected input[1, 9, 262, 262] to have 7 channels, but got 9 channels instead
在用自己生成的mask图片来测试stuctureFlow代码的时候,我们出现了上诉的报错。
这是因为输入的mask图片是3通道的RGB图,而程序需要的是1通道的灰度图,所以出现了这样的错误。
所以我们需要把“RGB”转成“L”。
from PIL import Image
mask = Image.open("./3_mask.png")
print(mask.mode) # 查看mask的类型"RGB" or "L"
out = mask.convert("L")
out.save("./mask3.png")