TF-Keras(旧版)
tf-keras
是 Keras 2.x 版本的名称。现在它作为单独的 GitHub 仓库托管 此处。虽然它是一个旧版框架,但 Hub 上仍然托管着 4500 多个模型。可以使用 huggingface_hub
库加载这些模型。您**必须**在您的机器上安装 tf-keras
或 keras<3.x
。
如果您有兴趣了解 Keras 3.x 支持,请查看 本指南。
安装完成后,您只需使用 from_pretrained_keras
方法即可从 Hub 加载模型。详细了解 from_pretrained_keras
此处。
from huggingface_hub import from_pretrained_keras
model = from_pretrained_keras("keras-io/mobile-vit-xxs")
prediction = model.predict(image)
prediction = tf.squeeze(tf.round(prediction))
print(f'The image is a {classes[(np.argmax(prediction))]}!')
# The image is a sunflower!
您也可以将您的 tf-keras
模型托管在 Hub 上。但是,请记住 tf-keras
是一个旧版框架。为了让更多用户使用,我们建议您使用 Keras 3.x 创建模型,并像上面描述的那样本地共享它。有关上传 tf-keras
模型的更多详细信息,请查看 push_to_hub_keras
文档。
from huggingface_hub import push_to_hub_keras
push_to_hub_keras(model,
"your-username/your-model-name",
"your-tensorboard-log-directory",
tags = ["object-detection", "some_other_tag"],
**model_save_kwargs,
)
其他资源
- 博客文章 将 Keras 放入 🤗 Hub 以进行协作训练和可重复性(2022 年 4 月)