PyTorch-gender-age-estimation icon indicating copy to clipboard operation
PyTorch-gender-age-estimation copied to clipboard

Published 20 hours ago • verified publisher icon Nebula4869
→

Metadata

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

PyTorch-gender-age-estimation

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

Environment

  • python==3.6.9
  • Pytorch==1.7.0
  • onnx==1.7.0
  • onnxruntime==1.5.2
  • opencv-python

Getting Started

  1. Configure parameters and dataset dictionary in "config.yaml".
  2. Run "main.py" to train the network.
  3. 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).
  4. Run "image_demo.py" and "video_demo.py" to test the onnx model with a single image or USB camera.

← Metadata

33
Stars
8
Forks
Watchers

Owner

verified publisher icon Nebula4869

Metadata

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

Back