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):
2万+



被折叠的 条评论
为什么被折叠?



