官方介绍
TensorFlow Hub是用于可重用机器学习模块的库
TensorFlow Hub是一个库,用于发布,发现和使用机器学习模型的可重用部分。甲模块是一个自包含的一块TensorFlow图表,以其重量和资产,可以跨不同的任务在被称为迁移学习的过程中重复使用沿。转移学习可以:
- 用较小的数据集训练模型,
- 提高泛化能力,以及
- 加快培训速度。
官方网址:
https://www.tensorflow.org/hub/
模块网址:
https://tfhub.dev/
安装
TensorFlow Hub需要通过pip命令安装
pip install tensorflow-hub
使用模块
比如可以使用 mobilenet 模块进行图片分类
使用方式和微调代码参见该模块网址