文章目录
- 项目介绍
- 查看 MobileNet V2 中的层
- 第一次尝试（失败）
- 第二次尝试（失败）
- 第三次尝试（成功）
- 完整代码
- 注意事项一
- 注意事项二
项目介绍
众所周知，MobileNet V3 模型是当下最优秀的轻量级网络模型之一，但想用 tf.keras.applications 迁移学习这个模型的时候，发现现在这个库还不支持 MobileNet V3，只支持 MobileNet V2。
而在学习 MobileNet V2 和 MobileNet V3 时，我发现它们最明显的区别就是有没有使用 SE 模块。
因此，我在想，能不能在迁移学习到的模型中的某几层之间添加新层呢？
为了方便演示，我们只取到 MobileNet V2 的第三个 Bl