tensorflow-hub
介绍
tfhub.dev包含一系列模型,根据处理对象Problem domain
分成了四类:Image, Text, Vedio, Audio
模型格式
模型格式又可分为TF.js, TFLite, Coral
TF.js
是用于浏览器的模型。refTFLite
使用的思路主要是从预训练的模型转换为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中使用此模型,并且提供了模型的使用方法,非常方便!!!!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。