Bottleneck结构(附代码)

本文介绍了ResNet网络中的Bottleneck结构,该结构通过1x1卷积减少通道数,接着使用3x3卷积进行特征提取,最后再用1x1卷积恢复通道数,从而在增加网络深度的同时减少了计算复杂度。Bottleneck设计在保持模型准确性的同时,有效控制了计算资源的消耗。代码展示了BottleneckBlock的实现细节,包括批量归一化、卷积和残差连接等操作。

Bottleneck结构出现在resnet中,参考paper
结构如下
在这里插入图片描述
左边是一个resnet block,右边是一个和它时间复杂度类似的bottleneck block。

bottleneck由三层组成:1x1, 3x3, 1x1,其中1x1负责缩小放大channel。

为什么要用这个结构呢,paper里是这样说的:
一般来说,增加网络的深度可以提高accuracy, 但是会增加计算量,而bottleneck既增加了深度,计算量又可以得到节约。

参考代码

class Bottleneck2D(nn.Module):
    expansion = 2

    def __init__(self, inplanes, planes, stride=1, downsample=None):
     
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

蓝羽飞鸟

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值