multi-task-learning-for-face

Smile - Emotion - Gender - Age Multi-task learning using BKNet / ResNet

Installation

  pip install -r requirements.txt

How to use it

0. First clone this repo

  git clone https://github.com/baoanh1310/multi-task-learning-for-face.git
  cd multi-task-learning-for-face

1. Try real time video stream

  python video_stream_demo.py --prototxt ./face_detector/deploy.prototxt.txt --model ./face_detector/res10_300x300_ssd_iter_140000.caffemodel

2. Test on input images using MTCNN for detect faces

  cd 4class_age
  python image_demo_mtcnn.py --image img.jpg

Optional

You can try ResNet model by following below steps:

  cd multi_resnet
  python video_stream_demo.py --prototxt ../face_detector/deploy.prototxt.txt --model ../face_detector res10_300x300_ssd_iter_140000.caffemodel