码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow-hub 介绍

时间:2021-04-21 11:56:03      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:log   def   notebook   开发   问题   服务   展示   copy   训练   

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中使用此模型,并且提供了模型的使用方法,非常方便!!!!

tensorflow-hub 介绍

标签:log   def   notebook   开发   问题   服务   展示   copy   训练   

原文地址:https://www.cnblogs.com/cnyogurt/p/14676331.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!