ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

tensorflow-hub 介绍

2021-04-19 13:02:09  阅读:216  来源: 互联网

标签:hub 模型 使用 介绍 格式 tensorflow ref 加载


tensorflow-hub介绍

scrnli_4_19_2021_9-54-50 AM.png

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

模型格式

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

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

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

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


scrnli_4_19_2021_10-35-24 AM.png
Collection是处理一类问题的一系列模型的集合,而且在Collection中会展示模型的性能(推理时间等!)在选择模型时非常方便
scrnli_4_19_2021_10-36-43 AM.png


模型使用

进入模型介绍页面后,如下图:
scrnli_4_19_2021_10-46-27 AM.png

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

标签:hub,模型,使用,介绍,格式,tensorflow,ref,加载
来源: https://www.cnblogs.com/cnyogurt/p/14676331.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有