Function build_transformer_decoder

WeVisionOne/transformer_decoder/build.py:9–14

Build a instance embedding branch from `cfg.MODEL.INS_EMBED_HEAD.NAME`.

(cfg, in_channels, mask_classification=True)

def build_transformer_decoder(cfg, in_channels, mask_classification=True):
    """
    Build a instance embedding branch from `cfg.MODEL.INS_EMBED_HEAD.NAME`.
    """
    name = cfg.MODEL.TRANSFORMER_DECODER.NAME
    return TRANSFORMER_DECODER_REGISTRY.get(name)(cfg, in_channels, mask_classification)