maskrcnn_benchmark理解记录——modeling\backbone\resnet.py

 

# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
"""
将cfg作为参数的resnet模块的变体。
Variant of the resnet module that takes cfg as an argument.
Example usage. Strings may be specified in the config file.

注意！！！！！！可以在配置文件中指定字符串。有两种组合。
StemWithGN是指stem层用group_norm作为归一化方式，替代BN。效果会好一些

    model = ResNet(
        "StemWithFixedBatchNorm",
        "BottleneckWithFixedBatchNorm",
        "ResNet50StagesTo4",
    )
OR:
    model = ResNet(
        "StemWithGN",
        "BottleneckWithGN",
        "ResNet50StagesTo4",
    )
Custom implementations may be written in user code and hooked in via the
`register_*` functions.
"""
from collections import namedtuple

import torch
import torch.nn.functional as F
from torch import nn

from maskrcnn_benchmark.layers import FrozenBatchNorm2d
from maskrcnn_benchmark.layers import Conv2d
from maskrcnn_benchmark.modeling.make_layers import group_norm
from maskrcnn_benchmark.utils.registry import Registry


# ResNet stage specification
StageSpec = namedtuple(
    "StageSpec",
    [
        "index",            # Index of the stage, eg 1, 2, ..,. 5    每层索引
        "block_count",      # Numer of residual blocks in the stage  每层residual blocks的个数
        "return_features",  # True => return the last feature map from this stage   是否返回此层的特征作为特征图。（特征提取层）
    ],
)

# -----------------------------------------------------------------------------
# Standard ResNet models
# -----------------------------------------------------------------------------
# ResNet-50 (including all stages)
#那么结合一下就是第二层conv2_x有3个残差块；第三层conv3_x有4个残差块；第四层conv4_x有6个残差块；第五层conv5_x有3个残差块；
#就是表格里的conv2_x →conv5_x   3;4;6;3。 并且返回最后一层作为特征提取层
ResNet50StagesTo5 = tuple(
    StageSpec(index=i, block_count=c, return_features=r)
    for (i, c, r) in ((1, 3, False), (2, 4, False), (3, 6, False), (4, 3, True))
)
# ResNet-50 up to stage 4 (excludes stage 5)
ResNet50StagesTo4 = tuple(
    StageSpec(index=i, block_count=c, return_features=r)
    for (i, c, r) in ((1, 3, False), (2, 4, False), (3, 6, True))
)
# ResNet-101 (including all stages)
ResNet101StagesTo5 = tuple(
    StageSpec(index=i, block_count=c, return_features=r)
    for (i, c, r) in ((1, 3, False), (2, 4, False), (3, 23, False), (4, 3, True))
)
# ResNet-101 up to stage 4 (excludes stage 5)
ResNet101StagesTo4 = tuple(
    StageSpec(index=i, block_count=c, return_features=r)
    for (i, c, r) in ((1, 3, False), (2, 4, False), (3, 23, True))
)
# ResNet-50-FPN (including all stages)  #todo this is my body
#conv2_x →conv5_x   3;4;6;3。
# conv2_x →conv5_x  作为特征提取层  那么对应POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125) 其实是第2到5层的池化 1/4；1/8;1/16；1/32
ResNet50FPNStagesTo5 = tuple(
    StageSpec(index=i, block_count=c, return_features=r)
    for (i, c, r) in ((1,