| license: apache-2.0 | |
| tags: | |
| - image-classification | |
| - pytorch | |
| metrics: | |
| - accuracy | |
| 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 | |
|  | |
| #### male | |
|  |