PaddleGAN
PaddleGAN copied to clipboard
EDVR模型因为tensor的shape不匹配报错
如果输入图片的高宽不是4的倍数,在746行代码(路径:ppgan/models/generator/evdr.py)
L3_fea = L3_fea.reshape([-1, N, self.nf, H // 4, W // 4])
会出现reshape失败的问题。
把图片的高宽均设为4的倍数就不会报错。
因为代码中会对图像进行2次下采样,所以要求输入图片的高宽为4的倍数~