Hello,I test your CelebA model using code
"python test.py \
--checkpoints ./checkpoints/celeba
--input ./examples/celeba /images
--mask ./examples/celeba /masks
--output ./checkpoints/results"
and it reports errors
loading error: ./examples/celeba/images/celeba_01.png
self.to_tensor(img_gray)
img_t = F.to_tensor(img).float()
img = img.view(pic.size[1], pic.size[0], nchannel)
RuntimeError: shape '[256, 256, 1]' is invalid for input of size 262144
Hello,I test your CelebA model using code "python test.py \ --checkpoints ./checkpoints/celeba --input ./examples/celeba /images --mask ./examples/celeba /masks --output ./checkpoints/results"
and it reports errors loading error: ./examples/celeba/images/celeba_01.png self.to_tensor(img_gray) img_t = F.to_tensor(img).float() img = img.view(pic.size[1], pic.size[0], nchannel) RuntimeError: shape '[256, 256, 1]' is invalid for input of size 262144
I don't know how to solve. thanks you!