mobilenet v1结构介绍

tony365

已于 2024-04-08 13:13:18 修改

阅读量397

点赞数 14

分类专栏： pytorch 文章标签： python pytorch

于 2024-04-08 10:50:36 首次发布

本文链接：https://blog.csdn.net/tywwwww/article/details/137499611

版权

本文详细介绍了MobileNetV1中使用的深度可分离卷积模块的实现，包括其结构、代码示例以及计算量分析。作者构建了一个MobilnetV1Block类，展示了如何构建并执行这种轻量级网络结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

- mobilenet v1

mobilenet v1

mobilenetv1中的深度可分离卷积模块

在这里插入图片描述
mobilenetv1完整结构：

计算量

在这里插入图片描述

参考：MobileNetv1论文详解.md

代码：

import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision.models as models
from torch import flatten

class MobilnetV1Block(nn.Module):
    """Depthwise conv + Pointwise conv"""

    def __init__(self, in_channels, out_channels, stride=1):
        super(MobilnetV1Block, self).__init__()
        # dw conv kernel shape is (in_channels, 1, ksize, ksize)
        self.dw = nn.Sequential(
            nn.Conv2d(in_channels, 64, kernel_size=3,
                      stride=stride, padding=1, groups=4, bias=False),
            nn.BatchNorm2d(in_channels),
            nn.