GoogLeNet的搭建

Miracle
Miracle
Miracle
1357
文章
109
评论
2018年10月13日10:24:43深度学习GoogLeNet的搭建已关闭评论2,26542903字阅读9分40秒阅读模式

废话不多说，上代码：文章源自联网快讯-https://x1995.cn/3334.html

 文章源自联网快讯-https://x1995.cn/3334.html

  1. import torch
  2. import torch.nn as nn
  3. #定义一个卷积加一个relu激活函数和一个batchnorm作为一个基本的层结构
  4. def conv_relu(in_channels, out_channels, kernel, stride=1, padding=0):
  5.      layer = nn.Sequential(
  6.           nn.Conv2d(in_channels, out_channels, kernel, stride, padding),
  7.           nn.BatchNorm2d(out_channels, eps=1e-3),
  8.           nn.ReLU(True)
  9.      )
  10.      return layer
  12. class inception(nn.Module):
  13.     def __init__(self, in_channel=3, out1_1=64, out2_1=48, out2_3=64, out3_1=64, out3_5=96, out4_1=32):
  14.           super(inception, self).__init__()
  15.           #第一条线路
  16.           self.branch1x1 = conv_relu(in_channel, out1_1, 1)
  18.           #第二条线路
  19.           self.branch3x3 = nn.Sequential(
  20.                conv_relu(in_channel, out2_1, 1),
  21.                conv_relu(out2_1, out2_3, 3, padding=1)
  22.           )
  24.           #第三条线路
  25.           self.branch5x5 = nn.Sequential(
  26.                conv_relu(in_channel, out3_1, 1),
  27.                conv_relu(out3_1, out3_5, 5, padding=2)
  28.           )
  30.           #第四条线路
  31.           self.branch_pool = nn.Sequential(
  32.                nn.MaxPool2d(3, stride=1, padding=1),
  33.                conv_relu(in_channel, out4_1, 1)
  34.           )
  35.     def forward(self, x):
  36.           f1 = self.branch1x1(x)
  37.           f2 = self.branch3x3(x)
  38.           f3 = self.branch5x5(x)
  39.           f4 = self.branch_pool(x)
  40.           output = torch.cat((f1, f2, f3, f4), dim=1)
  41.           return output
  43. class googlenet(nn.Module):
  44.     def __init__(self, verbose=False):
  45.           super(googlenet, self).__init__()
  46.           self.block1 = nn.Sequential(
  47.                conv_relu(3, out_channels=64, kernel=7, stride=2, padding=3),
  48.                nn.MaxPool2d(3, 2)
  49.           )
  50.           self.block2 = nn.Sequential(
  51.                conv_relu(64, 64, kernel=1),
  52.                conv_relu(64, 192, kernel=3, padding=1),
  53.                nn.MaxPool2d(3, 2)
  54.           )
  55.           self.block3 = nn.Sequential(
  56.                inception(192, 64, 96, 128, 16, 32, 32),
  57.                inception(256, 128, 128, 192, 32, 96, 64),
  58.                nn.MaxPool2d(3, 2)
  59.           )
  60.           self.block4 = nn.Sequential(
  61.                inception(480, 192, 96, 208, 16, 48, 64),
  62.                inception(512, 160, 112, 224, 24, 64, 64),
  63.                inception(512, 128, 128, 256, 24, 64, 64),
  64.                inception(512, 112, 144, 288, 32, 64, 64),
  65.                inception(528, 256, 160, 320, 32, 128, 128),
  66.                nn.MaxPool2d(3, 2)
  67.           )
  68.           self.block5 = nn.Sequential(
  69.                inception(832, 256, 160, 320, 32, 128, 128),
  70.                inception(832, 384, 182, 384, 48, 128, 128),
  71.                nn.AvgPool2d(2)
  72.           )
  74.           self.classifier = nn.Linear(1024, 10)
  76.     def forward(self, x):
  77.           x = self.block1(x)
  78.           x = self.block2(x)
  79.           x = self.block3(x)
  80.           x = self.block4(x)
  81.           x = self.block5(x)
  82.           x = x.view(x.size(0), -1)
  83.           x = self.classifier(x)
  84.           return x
  86. #可以看到输入的尺寸不断减小，通道的维度不断增加
  87. googlenet=googlenet(True)
  88. print(googlenet)
文章源自联网快讯-https://x1995.cn/3334.html
继续阅读
历史上的今天
10月
13
Miracle
  • 本文由 发表于 2018年10月13日10:24:43