- Age Estimation: I trained a regression model on UTKFace Dataset to estimate a person's age based on human facial appearance, using ResNet50 convolutional neural network
pip install -r requirements.txt
You can also download my age estimator model from here
Traning Information:
- Validation Loss: 5.6115
- Training Loss: 2.4291
python3 processing_on_image.py --img YOUR_IMAGE.JPG
python processing_on_image.py --help
For Example:
--model
: The age estimator model file path default:weights/best_age_estimator_resnet.h5
--draw, --no-draw
: You can draw bounding boxes on the image. default:True
--show, --no-show
: You can see the result by this command default:True
--save, --no-save
: You can save the result by this command default:True
python3 main.py --source YOUR_VIDEO_PATH OR CAMERA_ID
python main.py --help
For Example:
--model
: The age estimator model file path default:weights/best_age_estimator_resnet.h5
--draw, --no-draw
: You can draw bounding boxes on the image. default:True