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

How to run on custom input? #6

Closed Nerdyvedi closed 3 years ago

Nerdyvedi commented 3 years ago

Hi, I want to run the model on custom faces image. I could not find a parameter that takes in input images directory, It always runs the model on fixed input images.

Thanks

mahmoudnafifi commented 3 years ago

Hi,

Do you mean to run the model on real face images? If so, the code of this part is not published yet. I am busy with other stuff this month. I will do my best to post it soon.

The currently available code generates samples (for a specific domain) and accepts a target color histogram as input.

Nerdyvedi commented 3 years ago

@Nerdyvedi Could you let me know where should I make the changes, I can try and implemeting it on my own and then create a PR?

mahmoudnafifi commented 3 years ago

Check Sec. 2.3 in the paper. You are more than welcome to re-implement it. But just to let you know that the code of this part is already implemented but needs some polish that I am going to do soon before posting.