(cfg)
| 15 | |
| 16 | |
| 17 | def build_centernet_upsample_layer(cfg): |
| 18 | upsample_layer_name = cfg.MODEL.CENTERNET.UPSAMPLE_LAYER.NAME |
| 19 | if upsample_layer_name == '': ## currently, only for hg104 backbone |
| 20 | upsample_layer = nn.Identity() |
| 21 | upsample_layer.out_channels = 256 |
| 22 | else: |
| 23 | upsample_layer = UPSAMPLE_LAYER_REGISTRY.get(upsample_layer_name)(cfg) |
| 24 | return upsample_layer |
| 25 | |
| 26 | |
| 27 | def build_centernet_head(cfg, in_channels): |