'Save Image' in the demo is not working

switchablenorms / CelebAMask-HQ

A large-scale face dataset for face parsing, recognition, generation and editing.

2.09k stars 348 forks source link

'Save Image' in the demo is not working #76

Open Ha0Tang opened 3 years ago

opptimus commented 3 years ago

me too

Padmashreeg commented 3 years ago

The image is not saving in a proper format. Sample output has been added for your reference. Also, could you please specify the total_step parameter which you have used during training? According to the GitHub repository, total_step has a value set to 1000000. Please confirm. The output I have obtained is by setting total_step = 400000.

prraoo commented 3 years ago

One quick hack I found was to use the result and pass it the variable self.output_img