TF2.0中使用Tensorflow Hub实现迁移学习

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 则需要选在对应版本可用的模型

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值