The output is still incomplete
Closed this issue · 4 comments
iworldtong commented
python test.py -i imgs/ffhq_in.png -m imgs/ffhq_m.png -o ./imgs/example_output.jpg -c checkpoints/co-mod-gan-ffhq-9-025000.pth
doesn't work
zengxianyu commented
what input image and mask?
iworldtong commented
ffhq_in.png ffhq_m.png
python test.py -i imgs/ffhq_in.png -m imgs/ffhq_m.png -o ./imgs/ffhq_o.png -c checkpoints/co-mod-gan-ffhq-9-025000.pth
and result_img
zengxianyu commented
iworldtong commented
(I changed it to "parser.add_argument('--device', help='cpu|cuda', default='cuda:1')")
The result is correct, when using "parser.add_argument('--device', help='cpu|cuda', default='cuda:0')"