CNN中计算量FLOPs的计算

1、FLOPs的概念:全称是floating point operations per second,意指每秒浮点运算次数,即用来衡量硬件的计算性能;在CNN中用来指浮点运算次数;

2、计算过程:

如上,根据上图来计算第一层卷积层的FLOPs:

对于某个卷积层,它的FLOPs数量为:[(K_h * K_w * C_{in}) * C_{out} + C_{out}] * (H * W) = num\_params * (H * W),其中num\_params表示该层参数的数目。

这里AlexNet网络第一卷积层为例,它的FLOPs数目为: [(11 * 11 * 3) * 96 + 96] * (55 * 55) = 105,705,600

猜你喜欢

转载自www.cnblogs.com/zf-blog/p/10875782.html