face-analysis / emonet

Official implementation of the paper "Estimation of continuous valence and arousal levels from faces in naturalistic conditions", Antoine Toisoul, Jean Kossaifi, Adrian Bulat, Georgios Tzimiropoulos and Maja Pantic, Nature Machine Intelligence, 2021
https://www.nature.com/articles/s42256-020-00280-0
Other
260 stars 62 forks source link

Onnx model export! #9

Open andres-ramirez-duque opened 2 years ago

andres-ramirez-duque commented 2 years ago

Do you know someone who has tried to export emonet to tensorrt or onnx?

I have some problems with the format of the output (dict) and with the tensor indexing (negative indexes). Any suggestion? cheers AR