---
license: apache-2.0
tags:
- image-classification
- pytorch
model-index:
- name: gender-classification
  results:
  - task:
      name: Image Classification
      type: image-classification
    metrics:
    - name: Accuracy
      type: accuracy
      value: 0.970833
---

# gender-classification
Evaluate set: 240 unseen images, from 27/05-29/05 from ShotX, no duplicate, clean, 120 male and 120 female. 
Loss function is CrossEntropy.
Model finetuning on 1827 images from 15/05-21/05 from ShotX, base on [rizvandwiki/gender-classification](https://huggingface.co/rizvandwiki/gender-classification).
 . Using AdamW optimizer, weight_decay=0.05, CosineAnnealingLR scheduler, learning rate 5e-6, 20 epochs. 

        accuracy    loss

        0.970833    0.102212   


## Example Images


#### female

![female](images/female.jpg)

#### male

![male](images/male.jpg)