pet-segmentation / utils /h5_to_saved_pb.py
kongkip's picture
added saved_model to allow loading from with huggingface hub pretrained models
be719e5
raw
history blame
108 Bytes
import tensorflow as tf
model = tf.keras.models.load_model("tf_model.h5")
tf.saved_model.save(model, "")