search

hamzapehlivan / StyleRes

Other
68 stars 2 forks source link

Custom model integretion with StyleRes. #7

Closed waleedrazakhan92 closed 3 months ago

waleedrazakhan92 commented 4 months ago

Great work! I'm using StylreRes model to get a latent vector which i then use in my pipeline to edit images. I was hoping to integrate my custom StyleGan checkpoint instead of Nvidia's. In the paper it's mentioned that a fixed StyleGan generator is used in the training phase. And after the first phase the missing features are added by the Styleres model. I want to integrate my custom checkpoint instead of the pretrained one as the data that I've trained my model on is cleaner and different with perfect face front images. How do i integrate my own model and should there be any changes required to accomplish it? Your guidance would be highly appreciated

hamzapehlivan commented 3 months ago

Hi sorry for the late response,

I would first try to replace the existing generator with the new one to see if it works. Because both generators are trained with face images, StyleRes might work automatically.

If it does not work, I would retrain StyleRes with the new generator. Although training code is not available in the repo, I send it via email upon request.

All the best

waleedrazakhan92 commented 3 months ago

@hamzapehlivan Thankyou for the response. I'll try the suggested solution.