Instructions to use Stefaron/trash-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use Stefaron/trash-classifier with Keras:
# !pip install -U keras tensorflow huggingface_hub # Keras needs TensorFlow installed to read "hf://" paths, so the tensorflow backend is selected here; # "jax" and "torch" also work for computation once TensorFlow is installed. import os os.environ["KERAS_BACKEND"] = "tensorflow" import keras model = keras.saving.load_model("hf://Stefaron/trash-classifier") - Notebooks
- Google Colab
- Kaggle
Download config.json from Stefaron/trash-classifier: direct link, hf CLI and curl.
- Browser
- Download file 234 Bytes
-
https://huggingface.co/Stefaron/trash-classifier/resolve/main/config.json
- Command line
-
hf download hf://Stefaron/trash-classifier/config.json
-
curl -L -o config.json https://huggingface.co/Stefaron/trash-classifier/resolve/main/config.json
234 Bytes
| { | |
| "model_type": "resnet50", | |
| "pretrained": true, | |
| "num_classes": 6, | |
| "input_shape": [ | |
| 384, | |
| 384, | |
| 3 | |
| ], | |
| "learning_rate": 0.0001, | |
| "optimizer": "adam", | |
| "loss_function": "cross_entropy" | |
| } |