KerasHub: a pretrained modeling library
4 months ago
- #Keras
- #Machine Learning
- #Pretrained Models
- KerasHub是一个简单、灵活且快速的预训练模型库
- 提供Keras 3实现的流行模型架构，并附带Kaggle Models上的预训练检查点
- 模型可在TensorFlow、Jax和Torch后端上用于训练和推理
- KerasHub扩展了核心Keras API，以keras.layers.Layer和keras.Model形式提供组件
- 安装方式：稳定版`pip install --upgrade keras-hub` 或每日构建版`pip install --upgrade keras-hub-nightly`
- 当前安装会同时获取TensorFlow以使用tf.data API预处理，但训练可在任意后端运行
- 使用示例包括加载ResNet模型进行图像分类，或BERT模型微调IMDb影评
- 遵循语义化版本控制，但在0.y.z预发布阶段可能破坏兼容性
- 预训练模型按『原样』提供，不附带任何担保
- 学术引用请参考官方说明