Instructions to use diozhug/tensorflow-trash-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use diozhug/tensorflow-trash-classification with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://diozhug/tensorflow-trash-classification") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -13,10 +13,9 @@ There are 2 models available:
|
|
| 13 |
- trash-classification-no-aug.keras
|
| 14 |
- trash-classification-aug.keras
|
| 15 |
|
| 16 |
-
The `trash-classification-no-aug.keras` model trained
|
| 17 |
```python
|
| 18 |
datagen = ImageDataGenerator(
|
| 19 |
-
rescale=1./255,
|
| 20 |
validation_split=0.2
|
| 21 |
)
|
| 22 |
```
|
|
|
|
| 13 |
- trash-classification-no-aug.keras
|
| 14 |
- trash-classification-aug.keras
|
| 15 |
|
| 16 |
+
The `trash-classification-no-aug.keras` model trained without data augmentation:
|
| 17 |
```python
|
| 18 |
datagen = ImageDataGenerator(
|
|
|
|
| 19 |
validation_split=0.2
|
| 20 |
)
|
| 21 |
```
|