File size: 2,494 Bytes
6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc 24d29f6 6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc c6458c3 6ee59dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
---
library_name: keras
tags:
- Image Classification
language:
- en
metrics:
- accuracy
---
# cats-vs-dogs
## Model Overview
The cats-vs-dogs model is a convolutional neural network (CNN) trained on the [Cats vs Dogs dataset](https://www.tensorflow.org/datasets/catalog/cats_vs_dogs),
a large set of images of cats and dogs.
### Details
- **Size:** 3,453,121 parameters
- **Model type:** CNN
- **Optimizer:** `RMSprop`
- **Number of Epochs:**
- **Hardware:** Tesla V4
- **Emissions:** Not measured
- **Total Energy Consumption:** Not measured
### How to Use
```python
!pip install huggingface_hub["tensorflow"] -q
# We are loading our Keras model from the Hub via the `huggingface_hub.from_pretrained_keras`
from huggingface_hub import from_pretrained_keras
import tensorflow as tf
model = from_pretrained_keras("AiresPucrs/cats-vs-dogs")
model.trainable = False
model.summary()
```
## Intended Use
This model was created for research purposes only. We do not recommend any application of this model outside this scope.
## Performance Metrics
Accuracy
## Training Data
[cats_vs_dogs dataset](https://www.tensorflow.org/datasets/catalog/cats_vs_dogs).
The dataset consists of a set of images of cats and dogs.
**Details:**
- [Homepage](https://www.microsoft.com/en-us/download/details.aspx?id=54765)
- Version: 1.0
- Date Published: 5/9/2022
- File Size:786.7 MB
## Training hyperparameters
The following hyperparameters were used during training:
| Hyperparameters | Value |
| :-- | :-- |
| name | RMSprop |
| learning_rate | 9.999999747378752e-05 |
| decay | 0.0 |
| rho | 0.8999999761581421 |
| momentum | 0.0 |
| epsilon | 1e-07 |
| centered | False |
| training_precision | float32 |
## Limitations
We do not recommend using this model in real-world applications. It was solely developed for academic and educational purposes.
## Cite as
```latex
@misc{teenytinycastle,
doi = {10.5281/zenodo.7112065},
url = {https://github.com/Nkluge-correa/teeny-tiny_castle},
author = {Nicholas Kluge Corr{\^e}a},
title = {Teeny-Tiny Castle},
year = {2024},
publisher = {GitHub},
journal = {GitHub repository},
}
```
## License
The cats-vs-dogs pre-trained model is under the Apache 2.0 License.
The cats_vs_dogs Dataset is licensed under the [Creative Commons(CC) License CC BY-NC-SA 4.0](https://creativecommons.org/licenses/by/4.0/).
## Model Plot
<details>
<summary>View Model Plot</summary>
</details> |