Pytorch实现MobileNetV2神经网络搭建

最新推荐文章于 2024-08-10 08:27:38 发布

rock4you

最新推荐文章于 2024-08-10 08:27:38 发布

阅读量1.6k

点赞数 3

分类专栏： pytorch 计算机视觉论文阅读深度学习

本文链接：https://blog.csdn.net/rock4you/article/details/89409507

版权

本文档介绍如何使用Pytorch构建MobileNetV2模型。参照论文'MobileNetV2: Inverted Residuals and Linear Bottlenecks'，提供了模型的详细代码实现。

摘要由CSDN通过智能技术生成

论文全名 MobileNetV2: Inverted Residuals and Linear Bottlenecks

下载地址 https://arxiv.org/abs/1801.04381v4

Pytorch实现MobileNetV2神经网络搭建的代码：

#coding:utf-8
import torch
import torch.nn as nn
import torch.nn.functional as F 
import torchvision as tv 
import torchvision.transforms as transforms 
from torchvision.transforms import ToPILImage
from torch.autograd import Variable
from torch import optim
import os
import datetime

class Head(nn.Module):# MobileNet_2 网络的第1层
    def __init__(self):
        super(Head, self).__init__()
        self.conv = nn.Conv2d(3, 32, kernel_size=3, stride=2, padding=1, bias=False)
        self.bn   = nn.BatchNorm2d(32)
    def forward(self, x):
        out = F.relu6(self.bn(self.conv(x)))
        return out

class Tail(nn.Module): # MobileNet_2 网络的最后3层
    def __init__(self, num_class):