/hupu-score

HUPU auto-score network based on ResNet-50

Primary LanguagePython

hupu-score

HUPU auto-score network based on ResNet-50

Dataset

Hupu: 盒宫大会api

python crawl.py

SCUT: SCUT-FBP5500-Database

Model

Network: ResNet50 + FC

Loss: MAE (MAE >> MSE)

Env

pip install requests
pip install tensorflow-gpu==1.11
pip install face_recognition

Train

Fine-Tune: Fix ResNet & Train FC

python keras.py --mode fine_tune

Full-Train: Train ResNet & FC

python keras.py --mode train

Predict

python keras.py --mode predict --image test.jpg