Aneureka / animate-me

🍭 A miniapp to convert selfie to anime.
https://zhuanlan.zhihu.com/p/79389141
222 stars 19 forks source link

Pretrain model for Asian face #9

Open calvin886 opened 3 years ago

calvin886 commented 3 years ago

Dear Aneureka, Do you have the pretrain model for Asian faces that can share with us? since the official pretrain model is only good for white women, i cannot really get a good result when i put asian faces on it

Aneureka commented 3 years ago

Sorry I have not fine-tuned the model on asian faces cuz it causes too much resources and time... But I found some datasets for asian faces which may help: http://www.seeprettyface.com/mydataset_page2.html

calvin886 commented 3 years ago

Sorry I have not fine-tuned the model on asian faces cuz it causes too much resources and time... But I found some datasets for asian faces which may help: http://www.seeprettyface.com/mydataset_page2.html

Thank you so much, the datasets are really cool. btw, have you got a chance to estimate the FLOPS (the amount of calculation )of this UGATIT model?