danielroich / PTI

Official Implementation for "Pivotal Tuning for Latent-based editing of Real Images" (ACM TOG 2022) https://arxiv.org/abs/2106.05744
MIT License
897 stars 112 forks source link

opt multi image one time? #41

Open renrenzsbbb opened 2 years ago

renrenzsbbb commented 2 years ago

Thanks for your great work. I find that in multi_id_coach.py, you will opt one img per step. why can you not opt several img in batch per step. Another question, if I have one person with several pose img, how to get a best invision res. one img one ckpt or one ckpt for all img. Thank you in advance.

danielroich commented 1 year ago

Hi @renrenzsbbb , Regarding the first question - you can. Just an implementation difference. I would advise you to create one ckpt for all the images. Please see "mystyle: a personalized generative prior" which is a relatively new work based on PTI which does just that. They also released code "https://github.com/google/mystyle"