2022-04-02 20:01:06 +08:00
|
|
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
2022-04-22 15:22:43 +08:00
|
|
|
from easycv.utils.registry import Registry
|
2022-04-02 20:01:06 +08:00
|
|
|
|
|
|
|
|
MODELS = Registry('model')
|
|
|
|
|
BACKBONES = Registry('backbone')
|
|
|
|
|
NECKS = Registry('neck')
|
|
|
|
|
HEADS = Registry('head')
|
|
|
|
|
LOSSES = Registry('loss')
|
2022-10-24 17:20:12 +08:00
|
|
|
VOXEL_ENCODERS = Registry('voxel_encoder')
|
|
|
|
|
MIDDLE_ENCODERS = Registry('middle_encoder')
|
|
|
|
|
FUSION_LAYERS = Registry('fusion_layers')
|
|
|
|
|
TRANSFORMER = Registry('Transformer')
|
|
|
|
|
TRANSFORMER_LAYER = Registry('transformerLayer')
|
|
|
|
|
TRANSFORMER_LAYER_SEQUENCE = Registry('transformer-layers sequence')
|
|
|
|
|
POSITIONAL_ENCODING = Registry('position encoding')
|
|
|
|
|
ATTENTION = Registry('attention')
|
|
|
|
|
FEEDFORWARD_NETWORK = Registry('feed-forward Network')
|
