search

yuval-alaluf / restyle-encoder

Official Implementation for "ReStyle: A Residual-Based StyleGAN Encoder via Iterative Refinement" (ICCV 2021) https://arxiv.org/abs/2104.02699
https://yuval-alaluf.github.io/restyle-encoder/
MIT License
1.03k stars 155 forks source link

Release the pretrained of original pSp model on Car, Horse, AFHQ Wild, Church in the paper #46

Closed di-mi-ta closed 2 years ago

di-mi-ta commented 2 years ago

Hi,

This work is very useful!

Could you release the pre-trained of the original pSp model on Car, Horse, AFHQ Wild, Church reported in the main paper?

Thanks!

yuval-alaluf commented 2 years ago

I will try uploading the models to Google Drive. However I cannot guarantee that these models will work with the pSp code since the models were trained over 6-7 months ago.

di-mi-ta commented 2 years ago

Yes, thanks very much for your support. I will try to integrate your PSP pre-trained weights into the official pSp code.

di-mi-ta commented 2 years ago

Hi @yuval-alaluf, could you have the drive link for these models?

yuval-alaluf commented 2 years ago

I apologize for the delay. I haven't had the time to prepare everything. I hope it's still relevant. I have uploaded the 4 pSp models to the following Google Drive link:

https://drive.google.com/drive/folders/1-7DJzHlzzFqnD3frBhPlEipfbiH9tBPq?usp=sharing

I have also provided the encoder class that was used to train these models. As I mentioned before, there is no guarantee that these checkpoints will work with the code since they were trained a while back.