Function make_features

CV/Pytorch_classification/VGGNet/model.py:54–65

(cfg: list)

# 采用写配置文件的方式，将网络对应层的类型及其参数写入到sequential容器中
def make_features(cfg: list):
    # 空列表layers用于存放每一层创建的结构
    layers = []
    in_channels = 3
    for v in cfg:
        if v == "M":
            layers += [nn.MaxPool2d(kernel_size=2, stride=2)]
        else:
            conv2d = nn.Conv2d(in_channels, v, kernel_size=3, padding=1)
            layers += [conv2d, nn.ReLU(True)]
            in_channels = v
    return nn.Sequential(*layers)


cfgs = {