zengxianyu/co-mod-gan-pytorch

The output is still incomplete

Closed this issue · 4 comments

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

what input image and mask?

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

I got reasonable results by running this command

RJaMDJ.png

Maybe your pth file is corrupted in downloading. Could you download and try again?

(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')"