1.FLOPs的概念:全称是floating point operations per second,意指每秒浮点运算次数,即用来衡量硬件的计算性能:在CNN中用来指浮点运算次数: 2.计算过程: 如上,根据上图来计算第一层卷积层的FLOPs: 对于某个卷积层,它的FLOPs数量为:,其中表示该层参数的数目. 这里AlexNet网络第一卷积层为例,它的FLOPs数目为: .…
目录: 1.经典的卷积层是如何计算的 2.分析卷积层的计算量 3.分析卷积层的参数量 4.pytorch实现自动计算卷积层的计算量和参数量 1.卷积操作如下: http://cs231n.github.io/assets/conv-demo/index.html 假设卷积层的输入特征图大小为C_in x H x W, 卷积核大小为K_1 x K_2, padding = P_1 x P_2, stride = S_1 x S_2, filter 数目为C_out. 输出的特征图大小为C_out…