__init__.py 2.53 KB
Newer Older
Sugon_ldc's avatar
Sugon_ldc 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
# Copyright (c) OpenMMLab. All rights reserved.
from .augmentations import (AudioAmplify, CenterCrop, ColorJitter, Flip, Fuse,
                            Imgaug, MelSpectrogram, MultiScaleCrop, Normalize,
                            PytorchVideoTrans, RandomCrop, RandomRescale,
                            RandomResizedCrop, Resize, TenCrop, ThreeCrop,
                            TorchvisionTrans)
from .compose import Compose
from .formatting import (Collect, FormatAudioShape, FormatGCNInput,
                         FormatShape, ImageToTensor, JointToBone, Rename,
                         ToDataContainer, ToTensor, Transpose)
from .loading import (ArrayDecode, AudioDecode, AudioDecodeInit,
                      AudioFeatureSelector, BuildPseudoClip, DecordDecode,
                      DecordInit, DenseSampleFrames,
                      GenerateLocalizationLabels, ImageDecode,
                      LoadAudioFeature, LoadHVULabel, LoadLocalizationFeature,
                      LoadProposals, OpenCVDecode, OpenCVInit, PIMSDecode,
                      PIMSInit, PyAVDecode, PyAVDecodeMotionVector, PyAVInit,
                      RawFrameDecode, SampleAVAFrames, SampleFrames,
                      SampleProposalFrames, UntrimmedSampleFrames)
from .pose_loading import (GeneratePoseTarget, LoadKineticsPose,
                           PaddingWithLoop, PoseDecode, PoseNormalize,
                           UniformSampleFrames)

__all__ = [
    'SampleFrames', 'PyAVDecode', 'DecordDecode', 'DenseSampleFrames',
    'OpenCVDecode', 'MultiScaleCrop', 'RandomResizedCrop', 'RandomCrop',
    'Resize', 'Flip', 'Fuse', 'Normalize', 'ThreeCrop', 'CenterCrop',
    'TenCrop', 'ImageToTensor', 'Transpose', 'Collect', 'FormatShape',
    'Compose', 'ToTensor', 'ToDataContainer', 'GenerateLocalizationLabels',
    'LoadLocalizationFeature', 'LoadProposals', 'DecordInit', 'OpenCVInit',
    'PyAVInit', 'SampleProposalFrames', 'UntrimmedSampleFrames',
    'RawFrameDecode', 'DecordInit', 'OpenCVInit', 'PyAVInit',
    'SampleProposalFrames', 'ColorJitter', 'LoadHVULabel', 'SampleAVAFrames',
    'AudioAmplify', 'MelSpectrogram', 'AudioDecode', 'FormatAudioShape',
    'LoadAudioFeature', 'AudioFeatureSelector', 'AudioDecodeInit',
    'ImageDecode', 'BuildPseudoClip', 'RandomRescale',
    'PyAVDecodeMotionVector', 'Rename', 'Imgaug', 'UniformSampleFrames',
    'PoseDecode', 'LoadKineticsPose', 'GeneratePoseTarget', 'PIMSInit',
    'PIMSDecode', 'TorchvisionTrans', 'PytorchVideoTrans', 'PoseNormalize',
    'FormatGCNInput', 'PaddingWithLoop', 'ArrayDecode', 'JointToBone'
]