Downloads · 30 days
8
7% of all-time downloads
pokemon-image-identifier-151
sgtsaughter/pokemon-image-identifier-151
pokemon-image-identifier-151 is a machine learning model from sgtsaughter. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
The model was trained from this data set sgtsaughter/pokemon-classification-images-151
All-time downloads
109
Public
Repo size
9.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.tflite9.6 MB · 98%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Dec 24, 2024
- Updated
- Dec 24, 2024
- SHA
- 7a508a2d
Model card
From the Hugging Face model README
The model was trained from this data set sgtsaughter/pokemon-classification-images-151
I trained this model using google colab pro account using T4 GPU and High-RAM turned on (Go to Runtime -> Change runtime type in google colab to update this).
The training took about 30 to 40 minutes.
It uses MobileNetV2 as the base model
The .ipynb file assumes you're connecting your google drive, and in that drive had a folder called Dataset with a PokemonData.zip file within it.
The model uses early stopping to prevent overfitting.
It then goes through another round of fine-tuning afterwards.
It outputs a .tflite file for use on mobile devices.
- License
- apache-2.0
- Created
- Dec 24, 2024
- Updated
- Dec 24, 2024