Skip to main content
Donate to the Python Software Foundation or Purchase a PyCharm License to Benefit the PSF! Donate Now

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 :

png png png png png

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
democlassi-0.5-py3-none-any.whl (10.1 MB) Copy SHA256 hash SHA256 Wheel py3
democlassi-0.5.tar.gz (10.1 MB) Copy SHA256 hash SHA256 Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page