search

XingangPan / GAN2Shape

Code for GAN2Shape (ICLR2021 oral)
https://arxiv.org/abs/2011.00844
MIT License
575 stars 101 forks source link

Try on AFHQ dataset #27

Open yzliu567 opened 3 years ago

yzliu567 commented 3 years ago

Hi, your ideas on this work are very nice. I'm trying on afhq dog dataset. I've trained stylegan-ada on this dataset and get the inversion. Then I feed the data into your pipeline and use the default mvn light file to train your model. But the result is not good. Could you please offer me some suggestions? (I took out 10 images from the dataset to train.) image image image image

XingangPan commented 3 years ago

I suggesting first performing joint pretraining on more samples (like 200 samples), and then train for each instance. This is what we did for face data.

hito-Chen commented 3 years ago

@yzliu567 Hi, I did a similar study, and I want to ask you a question please. I've trained stylegan2-ada https://github.com/NVlabs/stylegan2-ada on my own datasets and get the model (.pkl). But I don't know how to get the inversion (.pt). I think you have solved it yet. Looking forward to your reply. Thanks!