Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 2.66 KB
Newer Older
unknown's avatar
Initial add code.  
unknown committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45 
# Copyright (c) OpenMMLab. All rights reserved.
from .backbones import (C3D, STGCN, X3D, MobileNetV2, MobileNetV2TSM, ResNet,
                        ResNet2Plus1d, ResNet3d, ResNet3dCSN, ResNet3dLayer,
                        ResNet3dSlowFast, ResNet3dSlowOnly, ResNetAudio,
                        ResNetTIN, ResNetTSM, TANet, TimeSformer)
from .builder import (BACKBONES, DETECTORS, HEADS, LOCALIZERS, LOSSES, NECKS,
                      RECOGNIZERS, build_backbone, build_detector, build_head,
                      build_localizer, build_loss, build_model, build_neck,
                      build_recognizer)
from .common import (LFB, TAM, Conv2plus1d, ConvAudio,
                     DividedSpatialAttentionWithNorm,
                     DividedTemporalAttentionWithNorm, FFNWithNorm,
                     SubBatchNorm3D)
from .heads import (ACRNHead, AudioTSNHead, AVARoIHead, BaseHead, BBoxHeadAVA,
                    FBOHead, I3DHead, LFBInferHead, SlowFastHead, STGCNHead,
                    TimeSformerHead, TPNHead, TRNHead, TSMHead, TSNHead,
                    X3DHead)
from .localizers import BMN, PEM, TEM
from .losses import (BCELossWithLogits, BinaryLogisticRegressionLoss, BMNLoss,
                     CBFocalLoss, CrossEntropyLoss, HVULoss, NLLLoss,
                     OHEMHingeLoss, SSNLoss)
from .necks import TPN
from .recognizers import (AudioRecognizer, BaseRecognizer, Recognizer2D,
                          Recognizer3D)
from .roi_extractors import SingleRoIExtractor3D
from .skeleton_gcn import BaseGCN, SkeletonGCN

__all__ = [
    'BACKBONES', 'HEADS', 'RECOGNIZERS', 'build_recognizer', 'build_head',
    'build_backbone', 'Recognizer2D', 'Recognizer3D', 'C3D', 'ResNet', 'STGCN',
    'ResNet3d', 'ResNet2Plus1d', 'I3DHead', 'TSNHead', 'TSMHead', 'BaseHead',
    'STGCNHead', 'BaseRecognizer', 'LOSSES', 'CrossEntropyLoss', 'NLLLoss',
    'HVULoss', 'ResNetTSM', 'ResNet3dSlowFast', 'SlowFastHead', 'Conv2plus1d',
    'ResNet3dSlowOnly', 'BCELossWithLogits', 'LOCALIZERS', 'build_localizer',
    'PEM', 'TAM', 'TEM', 'BinaryLogisticRegressionLoss', 'BMN', 'BMNLoss',
    'build_model', 'OHEMHingeLoss', 'SSNLoss', 'ResNet3dCSN', 'ResNetTIN',
    'TPN', 'TPNHead', 'build_loss', 'build_neck', 'AudioRecognizer',
    'AudioTSNHead', 'X3D', 'X3DHead', 'ResNet3dLayer', 'DETECTORS',
    'SingleRoIExtractor3D', 'BBoxHeadAVA', 'ResNetAudio', 'build_detector',
    'ConvAudio', 'AVARoIHead', 'MobileNetV2', 'MobileNetV2TSM', 'TANet', 'LFB',
    'FBOHead', 'LFBInferHead', 'TRNHead', 'NECKS', 'TimeSformer',
    'TimeSformerHead', 'DividedSpatialAttentionWithNorm',
    'DividedTemporalAttentionWithNorm', 'FFNWithNorm', 'ACRNHead', 'BaseGCN',
    'SkeletonGCN', 'CBFocalLoss', 'SubBatchNorm3D'
]