本文整理匯總了Python中maskrcnn_benchmark.modeling.registry.BACKBONES屬性的典型用法代碼示例。如果您正苦於以下問題：Python registry.BACKBONES屬性的具體用法？Python registry.BACKBONES怎麽用？Python registry.BACKBONES使用的例子？那麽, 這裏精選的屬性代碼示例或許可以為您提供幫助。您也可以進一步了解該屬性所在類maskrcnn_benchmark.modeling.registry的用法示例。
在下文中一共展示了registry.BACKBONES屬性的4個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: build_res2net_fpn_p3p7_backbone
# 需要導入模塊: from maskrcnn_benchmark.modeling import registry [as 別名]
# 或者: from maskrcnn_benchmark.modeling.registry import BACKBONES [as 別名]
def build_res2net_fpn_p3p7_backbone(cfg):
body = res2net.Res2Net(cfg)
in_channels_stage2 = cfg.MODEL.RESNETS.RES2_OUT_CHANNELS
out_channels = cfg.MODEL.RESNETS.BACKBONE_OUT_CHANNELS
in_channels_p6p7 = in_channels_stage2 * 8 if cfg.MODEL.RETINANET.USE_C5 \
else out_channels
fpn = fpn_module.FPN(
in_channels_list=[
0,
in_channels_stage2 * 2,
in_channels_stage2 * 4,
in_channels_stage2 * 8,
],
out_channels=out_channels,
conv_block=conv_with_kaiming_uniform(
cfg.MODEL.FPN.USE_GN, cfg.MODEL.FPN.USE_RELU
),
top_blocks=fpn_module.LastLevelP6P7(in_channels_p6p7, out_channels),
)
model = nn.Sequential(OrderedDict([("body", body), ("fpn", fpn)]))
model.out_channels = out_channels
return model
# def build_backbone(cfg):
# assert cfg.MODEL.BACKBONE.CONV_BODY in registry.BACKBONES, \
# "cfg.MODEL.BACKBONE.CONV_BODY: {} are not registered in registry".format(
# cfg.MODEL.BACKBONE.CONV_BODY
# )
# return registry.BACKBONES[cfg.MODEL.BACKBONE.CONV_BODY](cfg) 示例2: build_backbone
# 需要導入模塊: from maskrcnn_benchmark.modeling import registry [as 別名]
# 或者: from maskrcnn_benchmark.modeling.registry import BACKBONES [as 別名]
def build_backbone(cfg):
assert cfg.MODEL.BACKBONE.CONV_BODY in registry.BACKBONES, \
"cfg.MODEL.BACKBONE.CONV_BODY: {} are not registered in registry".format(
cfg.MODEL.BACKBONE.CONV_BODY
)
return registry.BACKBONES[cfg.MODEL.BACKBONE.CONV_BODY](cfg) 示例3: test_build_backbones
# 需要導入模塊: from maskrcnn_benchmark.modeling import registry [as 別名]
# 或者: from maskrcnn_benchmark.modeling.registry import BACKBONES [as 別名]
def test_build_backbones(self):
''' Make sure backbones run '''
self.assertGreater(len(registry.BACKBONES), 0)
for name, backbone_builder in registry.BACKBONES.items():
print('Testing {}...'.format(name))
if name in BACKBONE_CFGS:
cfg = load_config(BACKBONE_CFGS[name])
else:
# Use default config if config file is not specified
cfg = copy.deepcopy(g_cfg)
backbone = backbone_builder(cfg)
# make sures the backbone has `out_channels`
self.assertIsNotNone(
getattr(backbone, 'out_channels', None),
'Need to provide out_channels for backbone {}'.format(name)
)
N, C_in, H, W = 2, 3, 224, 256
input = torch.rand([N, C_in, H, W], dtype=torch.float32)
out = backbone(input)
for cur_out in out:
self.assertEqual(
cur_out.shape[:2],
torch.Size([N, backbone.out_channels])
) 示例4: build_backbone
# 需要導入模塊: from maskrcnn_benchmark.modeling import registry [as 別名]
# 或者: from maskrcnn_benchmark.modeling.registry import BACKBONES [as 別名]
def build_backbone(cfg):
"""
For Generalized_RCNN
"""
assert cfg.MODEL.BACKBONE.CONV_BODY in registry.BACKBONES, \
"cfg.MODEL.BACKBONE.CONV_BODY: {} are not registered in registry".format(
cfg.MODEL.BACKBONE.CONV_BODY
)
return registry.BACKBONES[cfg.MODEL.BACKBONE.CONV_BODY](cfg)