卷积层参数量和计算量计算公式

本文详细介绍了卷积神经网络中卷积层的参数量和计算量的计算方法。首先,定义了输入特征图、卷积核、滤波器的尺寸以及padding和stridding等关键参数。接着,分别计算了卷积层的参数量,包括卷积核参数、滤波器参数和bias参数。最后,分析了卷积操作的乘法和加法计算量,指出两者相等且等于Cout×Hout×Wout×Cin×Hk×Wk次。该文对于理解卷积网络的复杂度和资源需求具有指导意义。


参考https://www.cnblogs.com/qinduanyinghua/p/11302618.html

0. 变量定义

假设输入的特征图(feature map)的shape为[Height, Weight, Channel],对应的字母表示为 [ H i n , W i n , C i n ] [ H_{in}, W_{in}, C_{in}] [Hin,Win,Cin]; 卷积核(kernel)的shape为 [ H k , W k ] [H_{k}, W_{k}] [Hk,Wk], 滤波器(filter)的shape为 [ C o u t , H o u t , W o u t , C i n ] [C_{out}, H_{out}, W_{out}, C_{in}] [Cout,Hout,Wout,Cin]; 输出的feature map的shape为 [ H o u t , W o u t , C o u t ] [H_{out}, W_{out}, C_{out}] [Hout,Wout,Cout];padding= [ P 1 , P 2 ] [P_1, P_2] [P1,P

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值