search

mahmoudnafifi / HistoGAN

Reference code for the paper HistoGAN: Controlling Colors of GAN-Generated and Real Images via Color Histograms (CVPR 2021).
MIT License
272 stars 27 forks source link

Getting weird output #5

Closed Nerdyvedi closed 3 years ago

Nerdyvedi commented 3 years ago

Hi, I am not able to replicate the results correctly.

How do I get the final output?

For example, I used the following code python histoGAN.py --name Faces_histoGAN --generate True --target_his ./target_images/1.jpg --gpu 0

And got this output which seems to be noise generated-1-05-30-2021_10-13-13-ema

How can I use this to generate face samples as shown in the repo?

Thank you

mahmoudnafifi commented 3 years ago

It seems either you didn't download the pre-trained model or the path of the model is not correct. You should download the trained model and copy it in ./models/test_histoGAN/model_0.pt (it doesn't necessarily be 0, it can be any number. The code will take the highest number if more than one model exist).