Instructions to use keras/deit_base_distilled_patch16_224_imagenet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- KerasHub
How to use keras/deit_base_distilled_patch16_224_imagenet with KerasHub:
import keras_hub import keras # Load ImageClassifier model image_classifier = keras_hub.models.ImageClassifier.from_preset( "hf://keras/deit_base_distilled_patch16_224_imagenet", num_classes=2, ) # Fine-tune image_classifier.fit( x=keras.random.randint((32, 64, 64, 3), 0, 256), y=keras.random.randint((32, 1), 0, 2), ) # Classify image image_classifier.predict(keras.random.randint((1, 64, 64, 3), 0, 256))import keras_hub # Create a Backbone model unspecialized for any task backbone = keras_hub.models.Backbone.from_preset("hf://keras/deit_base_distilled_patch16_224_imagenet") - Keras
How to use keras/deit_base_distilled_patch16_224_imagenet with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://keras/deit_base_distilled_patch16_224_imagenet") - Notebooks
- Google Colab
- Kaggle
| { | |
| "module": "keras_hub.src.models.deit.deit_image_classifier_preprocessor", | |
| "class_name": "DeiTImageClassifierPreprocessor", | |
| "config": { | |
| "name": "dei_t_image_classifier_preprocessor", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_converter": { | |
| "module": "keras_hub.src.models.deit.deit_image_converter", | |
| "class_name": "DeiTImageConverter", | |
| "config": { | |
| "name": "dei_t_image_converter", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_size": [ | |
| 224, | |
| 224 | |
| ], | |
| "scale": [ | |
| 0.017124753831663668, | |
| 0.01750700280112045, | |
| 0.017429193899782133 | |
| ], | |
| "offset": [ | |
| -2.1179039301310043, | |
| -2.0357142857142856, | |
| -1.8044444444444445 | |
| ], | |
| "interpolation": "bicubic", | |
| "antialias": true, | |
| "crop_to_aspect_ratio": false, | |
| "pad_to_aspect_ratio": false, | |
| "bounding_box_format": "yxyx" | |
| }, | |
| "registered_name": "keras_hub>DeiTImageConverter" | |
| }, | |
| "config_file": "preprocessor.json" | |
| }, | |
| "registered_name": "keras_hub>DeiTImageClassifierPreprocessor" | |
| } |