PyTorch-gender-age-estimation

Train ResNet18 on AFAD dataset for gender and age estimation with Pytorch

Configure parameters and dataset dictionary in "config.yaml".
Run "main.py" to train the network.
Run "export.py" to convert the .pt model into onnx format. (a trained model with 93.14% validation gender estimation accuracy has been placed and converted already).
Run "image_demo.py" and "video_demo.py" to test the onnx model with a single image or USB camera.

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
logs-2020-11-20-14-37		logs-2020-11-20-14-37
models-2020-11-20-14-37		models-2020-11-20-14-37
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
config.yaml		config.yaml
dataset.py		dataset.py
export.py		export.py
image_demo.py		image_demo.py
main.py		main.py
model.py		model.py
test.jpg		test.jpg
video_demo.py		video_demo.py

Provide feedback