tomas-gajarsky / facetorch

Python library for analysing faces using PyTorch
Apache License 2.0
479 stars 42 forks source link

Feature Request: Add a valnece-arousal model #58

Closed tawichi closed 8 months ago

tawichi commented 9 months ago

Hi. Thank you for maintaining this useful repository. I found a valence-arousal model (NeurIPS 2022) under an MIT license. They uploaded some pre-trained weights. Could you add this to Facetorch?

tomas-gajarsky commented 8 months ago

This valence arousal model was just added in facetorch version 0.4.0

tawichi commented 8 months ago

Thank you!! I appreciate it very much!

tomas-gajarsky commented 8 months ago

Thanks for your sponsorship contribution!

I have made some further adjustments. The default offset for the valence and arousal values is 0, but one can change it in the configuration of the postprocessor.