Tensorflow2.0 在迁移学习到的模型中间添加新层

98 篇文章 110 订阅
70 篇文章 34 订阅

项目介绍

众所周知,MobileNet V3 模型是当下最优秀的轻量级网络模型之一,但想用 tf.keras.applications 迁移学习这个模型的时候,发现现在这个库还不支持 MobileNet V3,只支持 MobileNet V2。

而在学习 MobileNet V2MobileNet V3 时,我发现它们最明显的区别就是有没有使用 SE 模块

因此,我在想,能不能在迁移学习到的模型中的某几层之间添加新层呢?

为了方便演示,我们只取到 MobileNet V2 的第三个 Block,在每个 Block 中的深度级卷积层后面添加 SE 模块。

查看 MobileNet V2 中的层

在项目开始之前,要先查看 MobileNet V2 中的层,然后根据这些层的名字来把需要的层提取出来:

mobile = tf.keras.applications
  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

cofisher

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值