AlexNet卷积层、池化层的输出图像尺寸计算

AlexNet2012年时计算机的算力还不足以支持这么深层的模型,所以AlexNet当时是把模型并行放在两个GPU上,而这篇文章是合并计算的。值得注意的是,AlexNet模型中输入图像是224×224×3。

AlexNet模型

先看模型代码:

net = nn.Sequential(

         nn.Conv2d(3, 96, kernel_size=11, stride=4, padding=2), nn.ReLU(),

         nn.MaxPool2d(kernel_size=3, stride=2),

         nn.Conv2d(96, 256, kernel_size=5, stride=1, padding=2), nn.ReLU(),

         nn.MaxPool2d(kernel_size=3, stride=2),

         nn.Conv2d(256, 384, kernel_size=3, stride=1, padding=1), nn.ReLU(),

         nn.Conv2d(384, 384, kernel_size=3, stride=1, padding=1), nn.ReLU(),

         nn.C

  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
AlexNet是一种经典的卷积神经网络架构,由多个卷积层池化层和全连接层组成。以下是AlexNet网络每层输出特征的大小: 1. 输入层:输入图像的大小可以根据数据集进行不同的设定,通常为224×224×3(长×宽×通道数)。 2. 卷积层1:包含96个大小为11×11×3的过滤器,步长为4,padding为0。经过卷积操作后,输出特征图的大小为55×55×96。 3. 池化层1:采用最大池化,窗口大小为3×3,步长为2。经过池化操作后,输出特征图的大小为27×27×96。 4. 卷积层2:包含256个大小为5×5×48的过滤器,步长为1,padding为2。经过卷积操作后,输出特征图的大小为27×27×256。 5. 池化层2:采用最大池化,窗口大小为3×3,步长为2。经过池化操作后,输出特征图的大小为13×13×256。 6. 卷积层3:包含384个大小为3×3×256的过滤器,步长为1,padding为1。经过卷积操作后,输出特征图的大小为13×13×384。 7. 卷积层4:包含384个大小为3×3×192的过滤器,步长为1,padding为1。经过卷积操作后,输出特征图的大小为13×13×384。 8. 卷积层5:包含256个大小为3×3×192的过滤器,步长为1,padding为1。经过卷积操作后,输出特征图的大小为13×13×256。 9. 池化层3:采用最大池化,窗口大小为3×3,步长为2。经过池化操作后,输出特征图的大小为6×6×256。 10. 全连接层1:包含4096个神经元。 11. 全连接层2:包含4096个神经元。 12. 输出层:根据具体问题的分类数量进行设置。 总结:AlexNet共有5个卷积层,3个池化层和3个全连接层。每个层的输出特征大小如上所述。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值