轻量级网络之mobilenet_v1 pytorch实现
轻量级网络之mobilenet_v1 pytorch实现
前言:前面讲解了mobilenet 实现在移动端或者嵌入式中的轻量级网络,本文使用pytorch 搭建mobilenet_v1网络。
一、Mobilenet_v1 网络结构
1.Mobilenet_v1 网络结构如图所示
由此我们可以得出mobilenet_v1的网络结构由标准卷积、深度可分离卷积、平均池化、全连接层组成。
2.标准卷积模块
由Conv + BN + Relu
class ConvBNReLU(nn.Sequential):
原创
2022-04-25 14:04:57 ·
5690 阅读 ·
5 评论