trimap和alpha图与网络结构不匹配 about alphagan-matting HOT 6 CLOSED

你试一试

     trimap_img = Image.open(trimap_path).convert('L')
     alpha_img = Image.open(alpha_path).convert('L')

from alphagan-matting.

这个也不行。。

from alphagan-matting.

你用的什么版本的pytorch。或许这个说法是对的output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]。

from alphagan-matting.

按照那个说法 input_dataset.py中的76-79行应该改为

   self.transform = transforms.Compose([
            transforms.ToTensor(),
            transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
        ])

但是依然是 不匹配的错误。我用的是pytorch 1.0版本。

from alphagan-matting.

你试试看最新的这次更新的input_dataset.py

from alphagan-matting.

我记得昨天就是这样改过的，也在这里发布了，怎么没了。不过这个问题可以按照你最新的input_dataset.py来修改，可以的。

from alphagan-matting.