TensorFlow-Hub

官方介绍

TensorFlow Hub是用于可重用机器学习模块的库

TensorFlow Hub是一个库,用于发布,发现和使用机器学习模型的可重用部分。甲模块是一个自包含的一块TensorFlow图表,以其重量和资产,可以跨不同的任务在被称为迁移学习的过程中重复使用沿。转移学习可以:

  • 用较小的数据集训练模型,
  • 提高泛化能力,以及
  • 加快培训速度。

官方网址:
https://www.tensorflow.org/hub/

模块网址:
https://tfhub.dev/

安装

TensorFlow Hub需要通过pip命令安装

pip install tensorflow-hub

使用模块

比如可以使用 mobilenet 模块进行图片分类

使用方式和微调代码参见该模块网址


原文出处:/show_1EF4hTM9aw7N.html