当前位置:   article > 正文

TF2.0中使用Tensorflow Hub实现迁移学习_"hub.load(esrgan-tf2/1\")"

"hub.load(esrgan-tf2/1\")"

1.安装tensorflow-hub: 使用pip或者anaconda都可以

      命令:pip install tensorflow-hub 

                 conda install tensorflow-hub

2.官网上的信息更新不及时,gitbub上的比较新,地址:https://github.com/tensorflow/hub

   tf-hub在tf1和tf2中的使用方式不同,

 在tf2.0中只能使用hub.load()(其实是tf.saved_model.load()的封装)和hub.KerasLayer(方便在Keras模型中使用)

3.使用语法简单:

   handle = "https://hub.tensorflow.google.cn/google/imagenet/mobilenet_v2_050_96/feature_vector/4"

   m = hub.load(handle)

   result =  m(input_tensor)

4. tf-hub使用连接下载模型文件存放在 "C:\Users\Administrator\AppData\Local\Temp\tfhub_modules\"

5.tf-hub官网:https://hub.tensorflow.google.cn/。如果使用的tensorflow版本是2.0 则需要选在对应版本可用的模型

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/532013
推荐阅读
相关标签
  

闽ICP备14008679号