Name		Name	Last commit message	Last commit date
parent directory ..
Age_Estimating_Pytorch.ipynb		Age_Estimating_Pytorch.ipynb
age-estimating-torch.pth		age-estimating-torch.pth
inference.py		inference.py
model.py		model.py
readme.md		readme.md
requirements.txt		requirements.txt
test.py		test.py
train.py		train.py

readme.md

Estimating Age Pytorch

git clone https://github.com/MaryamBoneh/DeepLearning-Course.git

Run train.py for start training. you should select your device(cpu is default).

like following command:

python train.py --device cuda

python test.py --device cuda --dataset dataset/test

Give the input photo with the --input argument.

python inference.py --device cuda --weight age-estimating-torch.pth --input test_images/image1.png

You can download utkface-new dataset from here.