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

Using your code on matlab. #22

Closed Edison12300 closed 2 years ago

Edison12300 commented 2 years ago

Hi, I'm a student and I'm trying to implement your GAN network on Matlab. I'm curious that is this possible to be done? And if it is, is it possible for you to provide the codes for the GAN layers?

Thanks!

mahmoudnafifi commented 2 years ago

Hi, I think Matlab recently offers more flexibility in designing and training deep models compared to previous Matlab versions. Regarding the GAN layer specifications, please check supplementary materials and the code for more details. Also, you can find a Matlab version of the histogram layer used in HistoGAN loss .. you may need to modify it a bit to match what is used in the HistoGAN paper.

Good luck!

Edison12300 commented 2 years ago

I see. Thanks for answering!