yuval-alaluf / SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754
https://yuval-alaluf.github.io/SAM/
MIT License
632 stars 151 forks source link

To preserve identity while training #27

Closed MetiCodePrivateLimited closed 3 years ago

MetiCodePrivateLimited commented 3 years ago

Which parameters do I need to play with to preserve user's identity while training the ageing model?

yuval-alaluf commented 3 years ago

As I mentioned in your other issues, I highly recommend reading the paper in which I explain all losses and their function. Basically, to preserve identity, you will want to play with the dedicated identity loss.