Skip to content

Latest commit

 

History

History
49 lines (35 loc) · 2.14 KB

README.md

File metadata and controls

49 lines (35 loc) · 2.14 KB
Raw

HSEmotionONNX Python Library for Facial Emotion Recognition

Downloads pypi package PWC

License

The code of HSEmotionONNX Python Library is released under the Apache-2.0 License. There is no limitation for both academic and commercial usage.

Installing

    python setup.py install

It is also possible to install it via pip:

    pip install hsemotion-onnx

Usage

    from hsemotion_onnx.facial_emotions import HSEmotionRecognizer
    model_name='enet_b0_8_best_afew'
    fer=HSEmotionRecognizer(model_name=model_name)
    emotion,scores=fer.predict_emotions(face_img,logits=False)

The following values of model_name parameter are supported:

  • enet_b0_8_best_vgaf
  • enet_b0_8_best_afew
  • enet_b0_8_va_mtl
  • enet_b2_8
  • enet_b2_7

The method predict_emotions returns both the string value of predicted emotions (Anger, Contempt, Disgust, Fear, Happiness, Neutral, Sadness, or Surprise) and scores at the output of the last layer. If the logits parameter is set to True (by default), the logits are returned, otherwise, the posterior probabilities are estimated from the logits using softmax.

The versions of this method for a batch of images are also available

    emotions,scores=fer.predict_multi_emotions(face_img_list,logits=False)

Complete usage examples are available in the demo folder. It is necessary to install mediapipe to run the demo script.

The details about training of the models are available in the main repository