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%