AnimalFacesCNN / README.md
michellewidjaja's picture
Update README.md
0316448 verified
|
raw
history blame contribute delete
No virus
899 Bytes
# Animal Faces CNN Model
This model is a CNN trained on the Kaggle Animal Face dataset to classify cats, dogs, and wildlife. It is based on the ResNet18 architecture.
## Training Results:
- Validation Accuracy: 99.32%
- Dataset: Cats, Dogs, and Wildlife Faces
- Number of Epochs: 10
| **Epoch** | **Loss** | **Validation Accuracy** |
|:---------:|:--------:|:-----------------------:|
| 1 | 0.2374 | 95.11% |
| 2 | 0.1716 | 95.86% |
| 3 | 0.1381 | 97.53% |
| 4 | 0.1250 | 97.48% |
| 5 | 0.1038 | 97.53% |
| 6 | 0.0867 | 97.61% |
| 7 | 0.0914 | 98.44% |
| 8 | 0.0656 | 99.09% |
| 9 | 0.0462 | 99.35% |
| 10 | 0.0432 | 99.32% |