/gender-classification-tensorflow

Gender Classification Using Convolutional Neural network

Primary LanguagePython

Gender prediction fine tuning

Source Blog

Requiremnets

look at requirements.txt

opencv
keras
tensorflow

To Train

to train = python fine-tune.py --train_dir "data/images/train" --val_dir "data/images/test"

to test with certain pics = python predict.py --model "model/vggface-ft_gender.h5" --image "a.jpg"

to test with web-cam = python cam.py