fitrialif / hyperface-with-squeezenet Goto Github PK
View Code? Open in Web Editor NEWThis project forked from sourabhvora/hyperface-with-squeezenet
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. DataGen2.py is the Data Generator used for reading multiple labels from json files. It is a modified version of Keras's default Data Generator. 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. The implementation is slightly different. The original HyperFace architecture is built on top of AlexNet while the implementation here uses another architecture called SqueezeNet.