tensorflow-hub 介绍

2023-10-02 16:59:10

`tensorflow-hub`介绍

tfhub.dev包含一系列模型，根据处理对象Problem domain分成了四类：Image, Text, Vedio, Audio

模型格式

模型格式又可分为TF.js, TFLite, Coral

TF.js 是用于浏览器的模型。ref
TFLite 使用的思路主要是从预训练的模型转换为tflite模型文件，拿到移动端部署。ref
Coral 是google发布在一款开发板，类似树莓派，专门用于跑神经网络模型。ref

上述三类模型都是比较特殊的格式（在电脑上或服务器上运行不使用上述3种格式！），模型列表中的右上角有标出格式。而实际上还有其他格式的模型（TF2.0 Saved Model, Hub module等），只有进入模型页面后才能看见。

Tensorflow Serving 使用的是TF2.0 Saved Model格式的模型！

Collection是处理一类问题的一系列模型的集合，而且在Collection中会展示模型的性能（推理时间等！）在选择模型时非常方便

模型使用

进入模型介绍页面后，如下图：

COPY URL 复制模型链接，在加载模型时，可直接使用hub.load("model url")的方式来加载，它会先自动下载模型到本地，再加载模型。具体看下面的notebook
download 下载模型到本地，加载模型时，可直接使用hub.load("/home/xxx/modlepath")，指定本地模型的路径。
**open colab notebook** 在colab中使用此模型，并且提供了模型的使用方法，非常方便！！！！