1.卷积层

1.1torch.nn.Conv2d()类式接口

1 torch.nn.Conv2d(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True))

参数:

  • in_channel:输入数据的通道数,例RGB图片通道数为3;
  • out_channel:输出数据的通道数,也就是kernel数量;
  • kernel_size: 卷积核大小,可以是int,或tuple;kernel_size=2,意味着卷积大小(2,2),kernel_size=(2,3),意味着卷积大小(2,3)即非正方形卷积
  •   stride:步长,默认为1,与kernel_size类似,stride=2,意味着步长上下左右扫描皆为2, stride=(2,3),左右扫描步长为2,上下为3;
  •   padding:零填充
1 import  torch
2 import  torch.nn as nn
3 
4 x=torch.rand(1,1,28,28)                                  #batch, channel, height, width
5 layer=nn.Conv2d(1,3,kernel_size=3,stride=2,padding=1)    #in_channel(和上面x中的channel数量一致), out_channel(kernel个数)
6 
7 out=layer.forward(x)               #或者直接out=layer(x)
8 print(out.shape)                   #torch.Size([1, 3, 14, 14]) 1指一张图片,3指三个kernel,14*14指图片大小
  • 第五行layer的相关属性:
  • layer.weight.shape = torch.Size([3,1,3,3])  第一个3指out_channel,即输出数据的通道数(kernel个数),第二个1指in_channel,即输入数据的通道数,这个值必须和x的channel数目一致。(Tip:前两个参数顺序是和Conv2d函数的前两个参数顺序相反的)
  • layer.bias.shape = torch.Size([3]) 这个3指kernel个数,不同的通道共用一个偏置。
  • out输出维度的计算:$(n+2p)*(n+2p) $ 与 $f*f$ 做卷积操作= $\left \lfloor \frac{n+2p}{s}+1 \right \rfloor *\left \lfloor \frac{n+2p}{s}+1 \right \rfloor $
  • n表示图片尺寸,p表示padding,f表示kernel尺寸,s表示slide步长,(28+2-3)/2+1=14.5,向下取整,即14

1.2F.conv2d()函数式接口

PyTorch里一般小写的都是函数式的接口,相应的大写的是类式接口。函数式的更加low-level一些,如果不需要做特别复杂的配置只要用类式接口即可。

1 import torch
 2 from torch.nn import functional as F
 3 
 4 #手动定义卷积核(weight)和偏置
 5 w = torch.rand(16, 3, 5, 5)                       #16种3通道的5*5卷积核
 6 b = torch.rand(16)  
 7 
 8 #定义输入样本
 9 x = torch.randn(1, 3, 28, 28)                     #1张3通道的28*28的图像
10 
11 #2D卷积得到输出
12 out = F.conv2d(x, w, b, stride=1, padding=1) 
13 print(out.shape)                                  #torch.Size([1, 16, 26, 26])
14 
15 out = F.conv2d(x, w, b, stride=2, padding=2) 
16 print(out.shape)                                  #torch.Size([1, 16, 14, 14])

2.池化层Pooling(下采样)

1 torch.nn.MaxPool2d(kernel_size, stride=None, padding=0, dilation=1, return_indices=False, ceil_mode=False)

参数:

  • kernel_size(int or tuple) - max pooling的窗口大小,
  • stride(int or tuple, optional) - max pooling的窗口移动的步长。默认值是kernel_size
  • padding(int or tuple, optional) - 输入的每一条边补充0的层数
  • dilation(int or tuple, optional) – 一个控制窗口中元素步幅的参数
  • return_indices - 如果等于True,会返回输出最大值的序号,对于上采样操作会有帮助
  • ceil_mode - 如果等于True,计算输出信号大小的时候,会使用向上取整,代替默认的向下取整的操作

接着上面卷积运算得到的1张16通道的14*14的图像,在此基础上做池化。

torch.MaxPool2d类式接口

1 x=out                                    #torch.Size([1, 16, 14, 14])
2 layer=nn.MaxPool2d(2,stride=2)           #池化层(池化核为2*2,步长为2),最大池化
3 
4 out=layer(x)           
5 print(out.shape)                         #torch.Size([1, 16, 7, 7])

F.avg_pool2d()函数式接口

1 x = out                                  #torch.Size([1, 16, 14, 14])
2 out = F.avg_pool2d(x, 2, stride=2)
3 print(out.shape)                         #torch.Size([1, 16, 7, 7])

Tip:池化后通道数不变。

torch.nn.AdaptiveAvgPool2d((1, 1))可以将图片尺寸变成任何想要的尺寸(通过元组的形式设定)

3.upsample(上采样)

change kernel里只有python3 pytorch kernel_size_ide

1 F.interpolate(input, size=None, scale_factor=None, mode='nearest', align_corners=None)

参数:

  • size(int):输出的spatial大小;
  • scale_factor(float): spatial尺寸的缩放因子;
  • mode(string):上采样算法:nearest,linear,bilinear,trilinear,area,默认nearest;
  • align_corners(bool, optional):如果align_corners=True,则对齐input和output的角点像素,只会对 mode=linear, bilinear 和 trilinear 有作用。默认是 False。
1 x = out                                                  #torch.Size([1, 16, 7, 7])
2 
3 out = F.interpolate(x, scale_factor=2, mode='nearest')   #向上采样,放大2倍,最近插值
4 print(out.shape)                                         #torch.Size([1, 16, 14, 14])

4.RELU激活函数

4.1torch.nn.RELU()类式接口

1 x = out                                     #torch.Size([1, 16, 14, 14])
2 
3 layer = nn.ReLU(inplace=True)               #ReLU激活,inplace=True表示直接覆盖掉ReLU目标的内存空间
4 out = layer(x)
5 print(out.shape)                            #torch.Size([1, 16, 14, 14])

4.2F.relu()函数式接口

1 x = out                   #torch.Size([1, 16, 14, 14])
2 
3 out = F.relu(x)
4 print(x.shape)            #torch.Size([1, 16, 14, 14])

5.Batch Norm

归一化可以使代价函数平均起来看更对称,使用梯度下降法更方便。

通常分为两步:调整均值、方差归一化

5.1Batch Norm

一个Batch的图像数据shape为[样本数N, 通道数C, 高度H, 宽度W],将其最后两个维度flatten,得到的是[N, C, H*W],标准的Batch Normalization就是在通道channel这个维度上进行移动,对所有样本的所有值求均值和方差,所以有几个通道,得到的就是几个均值和方差。

eg.[6, 3, 784]会生成[3],代表当前batch中每一个channel的特征均值,3个channel有3个均值和3个方差,只保留了channel维度,所以是[3]。

5.2Layer Norm

样本N的维度上滑动,对每个样本的所有通道的所有值求均值和方差,所以一个Batch有几个样本实例,得到的就是几个均值和方差。

eg.[6, 3, 784]会生成[6]

5.3Instance Norm

在样本N和通道C两个维度上滑动,对Batch中的N个样本里的每个样本n,和C个通道里的每个样本c,其组合[n, c]求对应的所有值的均值和方差,所以得到的是N*C个均值和方差。

5.4Batch Norm详解

输入数据是6张3通道784个像素点的数据,将其分到三个通道上,在每个通道上也就是[6, 784]的数据,然后分别得到和通道数一样多的统计数据均值μ和方差σ,将每个像素值减去μ除以σ也就变换到了接近N(0,1)的分布,后面又使用参数β和γ将其变换到接近N(β,γ)的分布。

change kernel里只有python3 pytorch kernel_size_ide_02

Tip:μ和σ只是样本中的统计数据,是没有梯度信息的,不过会保存在运行时参数里。而γ和β属于要训练的参数,他们是有梯度信息的。

正式计算:

change kernel里只有python3 pytorch kernel_size_ide_03

nn.BatchNorm1d()

1 import torch
 2 from torch import nn
 3 
 4 x = torch.rand(100, 16, 784)     #100张16通道784像素点的数据,均匀分布
 5 
 6 layer = nn.BatchNorm1d(16)       #传入通道数,因为H和W已经flatten过了,所以用1d
 7 out = layer(x)
 8 
 9 print(layer.running_mean)
10 #tensor([0.0499, 0.0501, 0.0501, 0.0501, 0.0501, 0.0502, 0.0500, 0.0499, 0.0499,
11 #        0.0501, 0.0500, 0.0500, 0.0500, 0.0501, 0.0500, 0.0500])
12 print(layer.running_var)
13 #tensor([0.9083, 0.9083, 0.9083, 0.9084, 0.9083, 0.9083, 0.9084, 0.9083, 0.9083,
14 #        0.9083, 0.9083, 0.9083, 0.9084, 0.9084, 0.9083, 0.9083])

Tip:layer.running_mean和layer.running_var得到的是全局的均值和方差,不是当前Batch上的,只不过这里只跑了一个Batch而已所以它就是这个Batch上的。

nn.BatchNorm2d()

1 import torch
 2 from torch import nn
 3 
 4 x = torch.rand(1, 16, 7, 7)     #1张16通道的7*7的图像
 5 
 6 layer = nn.BatchNorm2d(16)      #传入通道数(必须和上面的通道数目一致)
 7 out = layer(x)
 8 
 9 print(out.shape)                #torch.Size([1, 16, 7, 7])
10 print(layer.running_mean)        
11 print(layer.running_var)      
12 print(layer.weight.shape)       #torch.Size([16])对应上面的γ
13 print(layer.bias.shape)         #torch.Size([16])对应上面的β
14 print(vars(layer))              #查看网络中一个层上的所有参数
15 # {'training': True, 
16 #   '_parameters': 
17 #       OrderedDict([('weight', Parameter containing:
18 #                               tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.], requires_grad=True)), 
19 #                   ('bias', Parameter containing:
20 #                               tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], requires_grad=True))]), 
21 #   '_buffers': 
22 #       OrderedDict([('running_mean', tensor([0.0527, 0.0616, 0.0513, 0.0488, 0.0484, 0.0510, 0.0590, 0.0459, 0.0448, 0.0586, 0.0535, 0.0464, 0.0581, 0.0481, 0.0420, 0.0549])), 
23 #                   ('running_var', tensor([0.9089, 0.9075, 0.9082, 0.9079, 0.9096, 0.9098, 0.9079, 0.9086, 0.9081, 0.9075, 0.9052, 0.9081, 0.9093, 0.9075, 0.9086, 0.9073])), 
24 #                   ('num_batches_tracked', tensor(1))]), 
25 # '_backward_hooks': OrderedDict(), 
26 # '_forward_hooks': OrderedDict(), 
27 # '_forward_pre_hooks': OrderedDict(), 
28 # '_state_dict_hooks': OrderedDict(), 
29 # '_load_state_dict_pre_hooks': OrderedDict(), 
30 # '_modules': OrderedDict(), 
31 # 'num_features': 16, 
32 # 'eps': 1e-05, 
33 # 'momentum': 0.1, 
34 # 'affine': True, 
35 # 'track_running_stats': True}

Tip:

  • layer.weight和layer.bias是当前batch上的;
  • 如果在定义层时使用了参数affine=False,那么就是固定γ=1和β=0不自动学习,这时参数layer.weight和layer.bias将是None。

5.5Train和Test

类似于Dropout,Batch Normalization在训练和测试时的行为不同。

测试模式下,$\mu$和$\sigma ^{2}$使用训练集得到的全局$\mu$和$\sigma ^{2}$,归一化前调用layer.eval()设置Test模式。

change kernel里只有python3 pytorch kernel_size_2d_04