problem on testing your CelebA model

knazeri / edge-connect

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212

http://openaccess.thecvf.com/content_ICCVW_2019/html/AIM/Nazeri_EdgeConnect_Structure_Guided_Image_Inpainting_using_Edge_Prediction_ICCVW_2019_paper.html

Other

2.5k stars 530 forks source link

problem on testing your CelebA model #116

Open zhengbowei opened 4 years ago

zhengbowei commented 4 years ago

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!

anshen666 commented 4 years ago

你好，我最近也在跑这个代码。可以加你交流一下吗？我的微信：loveanshen 我的QQ：519838354 我的邮箱：519838354@qq.com 非常期待你百忙中的回复