项目介绍
众所周知,MobileNet V3 模型是当下最优秀的轻量级网络模型之一,但想用 tf.keras.applications 迁移学习这个模型的时候,发现现在这个库还不支持 MobileNet V3,只支持 MobileNet V2。
而在学习 MobileNet V2 和 MobileNet V3 时,我发现它们最明显的区别就是有没有使用 SE 模块。
因此,我在想,能不能在迁移学习到的模型中的某几层之间添加新层呢?
为了方便演示,我们只取到 MobileNet V2 的第三个 Block,在每个 Block 中的深度级卷积层后面添加 SE 模块。
查看 MobileNet V2 中的层
在项目开始之前,要先查看 MobileNet V2 中的层,然后根据这些层的名字来把需要的层提取出来:
mobile = tf.keras.applications