Instructions to use keras/efficientnet_b1_ft_imagenet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- KerasHub
How to use keras/efficientnet_b1_ft_imagenet with KerasHub:
import keras_hub import keras # Load ImageClassifier model image_classifier = keras_hub.models.ImageClassifier.from_preset( "hf://keras/efficientnet_b1_ft_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/efficientnet_b1_ft_imagenet") - Keras
How to use keras/efficientnet_b1_ft_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/efficientnet_b1_ft_imagenet") - Notebooks
- Google Colab
- Kaggle
| { | |
| "module": "keras_hub.src.models.efficientnet.efficientnet_image_classifier_preprocessor", | |
| "class_name": "EfficientNetImageClassifierPreprocessor", | |
| "config": { | |
| "name": "efficient_net_image_classifier_preprocessor", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_converter": { | |
| "module": "keras_hub.src.models.efficientnet.efficientnet_image_converter", | |
| "class_name": "EfficientNetImageConverter", | |
| "config": { | |
| "name": "efficient_net_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", | |
| "crop_to_aspect_ratio": true | |
| }, | |
| "registered_name": "keras_hub>EfficientNetImageConverter" | |
| }, | |
| "config_file": "preprocessor.json" | |
| }, | |
| "registered_name": "keras_hub>EfficientNetImageClassifierPreprocessor" | |
| } |