search

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

Getting Current Age of person #63

Closed Abhijeet-Bhilare closed 1 year ago

Abhijeet-Bhilare commented 1 year ago

Hello, I am trying to run inference_playground.ipynb on colab. I could run successfully to transform age from 0-100. But I am curious to know, how algorithm predicts current age of face? and want to get that age in inference.

If this is possible, please tell me the steps to get current age of face which we input to algorithm.

Best Regards, Abhijeet

yuval-alaluf commented 1 year ago

To get the age of a given face image, you can use the age classifier that was provided in the repo: https://github.com/yuval-alaluf/SAM#pretrained-models It is a VGG classifier from DEX and fine-tuned on the FFHQ-Aging dataset.