Open xiaoming3526 opened 4 years ago
Hello~ I have the same question with you . I wonder if you have solved the problem~
Hello~ I have the same question with you . I wonder if you have solved the problem~
Sorry, I didn't modify it successfully~we can talk about it
中国人说中文吧hahh 我微信号zihang_song------------------ 原始邮件 ------------------ 发件人: "xiaoming3526"notifications@github.com 发送时间: 2019年11月22日(星期五) 晚上6:17 收件人: "knazeri/edge-connect"edge-connect@noreply.github.com; 抄送: "songzihang"942737735@qq.com;"Comment"comment@noreply.github.com; 主题: Re: [knazeri/edge-connect] pre-trained models run errors (#130)
Hello~ I have the same question with you . I wonder if you have solved the problem~
Sorry, I didn't modify it successfully~we can talk about it
— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.
你好,我最近也在跑这个代码。可以加你交流一下吗?我的微信:loveanshen 我的QQ:519838354 我的邮箱:519838354@qq.com 非常期待你百忙中的回复
Hello~ I have the same question with you . I wonder if you have solved the problem~
你好,我最近也在跑这个代码。可以加你交流一下吗?我的微信:loveanshen 我的QQ:519838354 我的邮箱:519838354@qq.com 非常期待你百忙中的回复
i am also get this error
Loading InpaintingModel generator...
start testing...
Traceback (most recent call last):
File "test.py", line 2, in
when run this command in google colab
!python test.py --model 2 --checkpoints checkpoints/celeba_models --input examples/images --mask examples/masks --output checkpoints/results
I wonder if you have solved the problem. I have the same problem. Can you help me to solve it?
I solved the problem by changing the "torchvision" to the version "torchvision-0.2.2".
I downloaded the pre-training model to see the effect of the model. Instead of generating flist file, I directly tested it, and the following error occurred:
python test.py --checkpoints ./checkpoints/places2 --input ./examples/places2/images --mask ./examples/places2/masks --output ./checkpoints/results --model 3
Loading EdgeModel generator... Loading InpaintingModel generator...
start testing...
loading error: ./examples/places2/images/places2_01.png Traceback (most recent call last): File "/home/fanming_maming_cs/fm/edge-connect-master/src/dataset.py", line 41, in getitem item = self.load_item(index) File "/home/fanming_maming_cs/fm/edge-connect-master/src/dataset.py", line 83, in load_item return self.to_tensor(img), self.to_tensor(img_gray), self.to_tensor(edge), self.to_tensor(mask) File "/home/fanming_maming_cs/fm/edge-connect-master/src/dataset.py", line 155, in to_tensor img_t = F.to_tensor(img).float() File "/root/miniconda3/envs/edge-connect/lib/python3.6/site-packages/torchvision/transforms/functional.py", line 71, in to_tensor img = img.view(pic.size[1], pic.size[0], nchannel) RuntimeError: shape '[256, 256, 1]' is invalid for input of size 262144
Please help to check if there are any mistakes. Thank you!