程序记录

目标检测-pytorch

1、strides = [stride] + [1]*(num_blocks-1)

>>b=[1]+[1]*2
>>print(b)
[1, 1, 1]

def _make_layer(self, block, planes, num_blocks, stride):
    strides = [stride] + [1]*(num_blocks-1)
    layers = []
    for stride in strides:
       layers.append(block(self.in_planes, planes, stride))
       self.in_planes = planes * block.expansion
    return nn.Sequential(*layers)