图像卷积及其计算(特征图尺寸、参数量、计算量)
卷积前后特征图尺寸的计算
定义参数如下:
- 输入特征图尺寸: W×WW×WW×W
- 卷积核尺寸: F×FF×FF×F
- 步长: SSS
- 填充的像素数:PPP
则有输出特征图尺寸为 N×NN×NN×N:
N=(W−F+2P)/S+1N = (W − F + 2P )/S+1 N=(W−F+2P)/S+1
参数量的计算
卷积层的参数量
卷积的参数量即卷积核的参数量,设我们有如下参数:
- 卷积核尺寸:KKK
- 前一层的通道数:CinC_{in}Cin
- 当前层的卷积核个数:CoutC_{out}Cout
单个卷积核的参数量:
Paramskernel=Cin×K2Params_{kernel}=C_{in}\times K^2 Paramskernel=Cin×K2
有 CoutC_{out}Cout 个卷积核,故:
Paramsconv=K2×Cin×CoutParams_{conv} = K^2\times C_{in}\times C_{out} Paramsconv=K2×Cin×Cout
全连接层的参数量
上面已经说过卷积层的参数量计算方法了,那如何计算全连接层的参数量?其实和卷积层参数量的计算方法是一样的。我们在进行全连接层的计算之前需要将最后一层卷积得到的特征图展开为一维的向量,即 Din=H×W×CD_{in}=H\times W\times CDin=H×W×C ,其中 H,W,CH,\ W,\ CH, W, C 是最后一层卷积输出特征图的高宽和通道数,DinD_{in }Din 即为本全连接层的输入特征维度,又设 DoutD_{out}Dout 为输出特征维度,则有:
ParamsFC=Din×DoutParams_{FC}=D_{in}\times D_{out} ParamsFC=Din×Dout
可以理解为是一个卷积层,我们就是用4096个 7×7×5127\times 7\times 5127×7×512 的卷积核去做卷积。
验证
我们用VGG16, PyTorch来验证我们的公式是否正确。
VGG16的第一个卷积层输入为原图像 224×224×3224\times 224\times 3224×224×3,卷积核个数为 646464,尺寸为3,则根据公式,我们有VGG16第一个卷积层的参数量:
Paramsconv=K2×Cin×Cout=32×3×64=1728Params_{conv} = K^2\times C_{in}\times C_{out}= 3^2\times 3\times 64=1728 Paramsconv=K2×Cin×Cout=32×3×64=1728
VGG16第一个全连接层的输入维度数为最后一张特征图的尺寸拉直:Din=7×7×512D_{in}=7\times 7\times 512Din=7×7×512,输出维度数为4046,则:
ParamsFC=Din×Dout=4096×4096=102760448Params_{FC}=D_{in}\times D_{out}=4096\times 4096=102760448 ParamsFC=Din×Dout=4096×4096=102760448
手动计算好之后我们用PyTorch来打印输出以下看一下:
import torch
from torchvision.models import resnet50, vgg16
import numpy as npmodel = vgg16()
for name, parameters in model.named_parameters():print(name, ':', np.prod(parameters.size()))
输出:
features.0.weight : 1728
features.0.bias : 64
...
classifier.0.weight : 102760448
classifier.0.bias : 4096
计算正确。
卷积的计算量
参考:FLOPs、FLOPS、Params的含义及PyTorch中的计算方法
MAC
MAC:Multiply Accumulate,乘加运算。乘积累加运算(英语:Multiply Accumulate, MAC)是在数字信号处理器或一些微处理器中的特殊运算。实现此运算操作的硬件电路单元,被称为“乘数累加器”。这种运算的操作,是将乘法的乘积结果和累加器的值相加,再存入累加器:
a←a+b×ca\leftarrow a+b\times c a←a+b×c
使用MAC可以将原本需要的两个指令操作减少到一个指令操作,从而提高运算效率。
计算量FLOPs的计算
以下不考虑激活函数的计算量。
卷积层
(2×Ci×K2−1)×H×W×C0(2\times C_i\times K^2-1)\times H\times W\times C_0(2×Ci×K2−1)×H×W×C0
CiC_iCi=输入通道数, KKK=卷积核尺寸,H,WH,WH,W=输出特征图空间尺寸,CoC_oCo=输出通道数。
一个MAC算两个个浮点运算,所以在最前面×2\times 2×2。不考虑bias时有−1-1−1,有bias时没有−1-1−1。由于考虑的一般是模型推理时的计算量,所以上述公式是针对一个输入样本的情况,即batch size=1。
理解上面这个公式分两步,括号内是第一步,计算出输出特征图的一个pixel的计算量,然后再乘以 H×W×CoH\times W\times C_oH×W×Co 拓展到整个输出特征图。
括号内的部分又可以分为两步,(2⋅Ci⋅K2−1)=(Ci⋅K2)+(Ci⋅K2−1)(2\cdot C_i\cdot K^2-1)=(C_i\cdot K^2)+(C_i\cdot K^2-1)(2⋅Ci⋅K2−1)=(Ci⋅K2)+(Ci⋅K2−1)。第一项是乘法运算数,第二项是加法运算数,因为 nnn 个数相加,要加 n−1n-1n−1 次,所以不考虑bias,会有一个−1-1−1,如果考虑bias,刚好中和掉,括号内变为 2⋅Ci⋅K22\cdot C_i\cdot K^22⋅Ci⋅K2。
全连接层
全连接层: (2×I−1)×O(2\times I-1)\times O(2×I−1)×O
III=输入层神经元个数 ,OOO=输出层神经元个数。
还是因为一个MAC算两个个浮点运算,所以在最前面×2\times 2×2。同样不考虑bias时有−1-1−1,有bias时没有−1-1−1。分析同理,括号内是一个输出神经元的计算量,拓展到OOO了输出神经元。
可以用 thop
库来验证。
Ref
https://zhuanlan.zhihu.com/p/91277743