如何计算卷积神经网络参数

如何计算卷积神经网络参数

卷积层(Conv Layer)的输出张量的大小
  • W1、H1、D1——输入图像的宽、高、深度(通道数)
  • W2、H2、D2——输出卷积后特征图的宽、高、深度(通道数)
  • P——padding,即图像填充零的厚度
  • S——stride,步长
  • F——卷积核的尺寸
  • K——卷积核的个数
    W 2 = W 1 − F + 2 P S + 1 W2 = { {W1 - F + 2P} \above{1pt} S} + 1 W2=SW1F+2P+1
    H 2 = H 1 − F + 2 P S + 1 H2 = { {H1 - F + 2P} \above{1pt} S} + 1 H2=SH1F+2P+1
    D 2 = K D2 = K D2=K
池化层(MaxPool Layer)的输出张量的大小
  • W1、H1、D1——输入图像的宽、高、深度(通道数)
  • W2、H2、D2——输出卷积后特征图的宽、高、深度(通道数
  • S——stride,步长
  • F——卷积核的尺寸
    注:不同于卷积层,池化层的输出通道数不改变
    W 2 = W 1 − F S + 1 W2 = { {W1 - F} \above{1pt} S} + 1 W2=SW1F+1
    H 2 = H 1 − F S + 1 H2 = { {H1 - F} \above{1pt} S} + 1 H2=SH1F+1
    D 2 = D 1 D2 = D1 D2=D1

学习资料:
http://cs231n.stanford.edu/slides/2016/winter1516_lecture7.pdf

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值