Skip to content

Emotion Recognition using ONNX Runtime

Notifications You must be signed in to change notification settings

vcelak/Emotion_onnx

This branch is up to date with Shohruh72/Emotion_onnx:main.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

df4dbd0 · May 27, 2024

History

5 Commits
May 27, 2024
May 27, 2024
May 27, 2024
May 27, 2024

Repository files navigation

Emotion Recognition inference code using ONNX Runtime

Vizualization

Installation

conda create -n ONNX python=3.8
conda activate ONNX
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
pip install onnxruntime-gpu==1.14.0
pip install opencv-python==4.5.5.64

Features:

  • Real-time face detection using ONNX model.
  • Emotion recognition from detected faces using another ONNX model.
  • Visualization of emotion scores on the video stream.

Download the pre-trained ONNX models:

  • Place the downloaded models in the weights directory
Download weight

WebCam Inference

$ python demo.py

Note

  • This repo supports only inference, see reference for more details

Reference

About

Emotion Recognition using ONNX Runtime

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%