Collection of my python functions for training pytorch models to classify emotion, age, race, gender
Project description
DEmoClassi
DEmoClassi stands for Demographic (age, gender, race) and Emotions (happy, sad, angry, ...) Classification.
This repository is an attempt to tackle the task of predicting the facial expression, age gender and race prediction from facial images.
As I was planning to train the models on Google Colab platform, I decided to gather
all my utility functions/classes in a package I named democlassi which is pip
installable via pip install --upgrade democlassi
.
The Training is done using Pytorch and pytorch-ignite and more information about the data used, the training process and live prediction using pretrained models and opencv can be found on this page.
Some sample predictions :
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distribution
Hashes for democlassi-0.5.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 26ffecf4edcb91606d2d52131a00c9b305dd413291c0da4c539479aec96b28cf |
|
MD5 | a1e8ddc5acdb74d4fa7d9c2cfddfe6fd |
|
BLAKE2b-256 | 00da54752876637ff64befd1bd1585dd5fca5c134e2bb44eec611053123d4a1f |