Skin color classification model trained on FairFace dataset. Download the dataset and place it in the folder data/fairface
To train the network, run:
python3 train.py --config config/base.yaml
To inference a single image, run:
python3 inference.py --config config/base.yaml --model-path /path/to/model.pt --image-path samples/test1.jpg
Inference results will be saved to inference/
Pretrained model: model.pt