search

mahmoudnafifi / HistoGAN

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

problems for testing rehistoGAN #31

Closed z-shijie closed 1 year ago

z-shijie commented 1 year ago

Hi, I want to test rehistoGAN, I have pre-downloaded the model, but when I run it with the statement left in your README file,

python histoGAN.py --name Faces_histoGAN --generate True --target_his ./target_images/2.jpg --gpu 0

the image is displayed without input. I try to input the image myself, but it fails. What should I do? I look forward to your answer. Thanks.

ps(What is "--target_his ./target_images/2.jpg".Is it a folder?)

mahmoudnafifi commented 1 year ago

Please make sure the model is located in the right location. Check this issue: if your model is Faces_histoGAN, then you should have this folder structure: models-> Faces_histoGAN -> model_0.pt

z-shijie commented 1 year ago

Thank you very much for your answer. I have solved the problem.