一、卷积的计算细节
1. params的计算
首先,卷积核必须要初始化,在卷积计算过程中,卷积核的不同层都会对图片的每一个通道进行conv操作,所以然后将所有输入通道叠加起来。假设输入通道数是Cin,卷积核尺寸为KH,KW,输出尺寸为Cout,所以卷积的参数量是Cin×KH×KW×Cout, 假如计算bias中参数(Cin×KH×KW+1)×Cout。
2. 特征图尺寸的计算
假设padding尺寸为p,stride为s,则输出尺寸的计算公式为:
Hout = (Hin+2p-1)/s+1
3. FLOPs的计算
先捋清这个概念。
FLOPS:注意全大写,是floating point operations per second的缩写,意指每秒浮点运算次数,理解为计算速度。是一个衡量硬件性能的指标。FLOPs:注意s小写,是floating point operations的缩写(s表复数),意指浮点运算数,理解为计算量。可以用来衡量算法/模型的复杂度。
因为一个Operation有两个MAC(加法和乘法运算),即在计算时候先乘以