Skip to content

Two multitasking CNNs for simultaneous face detection, landmarks estimation and visibility, pose estimation and gender recognition. hyperface.py concatenates feature maps from initial, mid and final layers of the network and then branches out to different heads. multiout.py branches from the final layer of the network to different heads. DataGen…

Notifications You must be signed in to change notification settings

sourabhvora/HyperFace-with-SqueezeNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HyperFace-with-SqueezeNet

Two multitasking CNNs for simultaneous face detection, landmarks estimation and visibility, pose estimation and gender recognition. Implemented for the final course project of the Neural Networks and Pattern Recognition (CSE253) course at UCSD.

hyperface.py

It concatenates feature maps from initial, mid and final layers of the network and then branches out to different heads.  

multiout.py

It branches from the final layer of the network to different heads.

DataGen2.py

It is the Data Generator used for reading multiple labels from json files. It is a modified version of Keras's default Data Generator.

Sample Outputs

SampleOut1.JPG, SampleOut2.JPG, SampleOut3.JPG are the results of the hyperface model. On the left is the visualization of the predicted head pose of the person while on the right are the estimated landmarks, their visibility (the landmarks which the network predicts as occluded are shown in red) and the predicted gender.

Reference

The idea is based on the following paper - [1] R. Ranjan, V. M. Patel, and R. Chellappa. Hyperface: A deep multitask learning framework for face detection, landmark localization, pose estimation, and gender recognition. CoRR, abs/1603.01249, 2016.

This implementation is slightly different. The original HyperFace architecture is built on top of AlexNet while the implementation here uses another architecture called SqueezeNet.

About

Two multitasking CNNs for simultaneous face detection, landmarks estimation and visibility, pose estimation and gender recognition. hyperface.py concatenates feature maps from initial, mid and final layers of the network and then branches out to different heads. multiout.py branches from the final layer of the network to different heads. DataGen…

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages