itshemantkmr/2D_Shape_Image_Datasets
Viewer • Updated • 20k • 12 • 2
How to use itshemantkmr/2D_Shape_Image_Classification 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://itshemantkmr/2D_Shape_Image_Classification")
Base model
google/mobilenet_v2_1.0_224