CelebAMask-HQ
CelebAMask-HQ copied to clipboard
'Save Image' in the demo is not working
me too
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.
One quick hack I found was to use the result and pass it the variable self.output_img