mogen/datasets/pipelines/__init__.py

from .compose import Compose
from .formatting import (
    Collect,
    ToTensor,
    Transpose,
    WrapFieldsToLists,
    to_tensor
)
from .siamese_motion import ProcessSiameseMotion, SwapSiameseMotion
from .transforms import Crop, Normalize, RandomCrop
from .motionverse import (
    LoadMotion,
    RetargetSkeleton,
    MotionDownsample,
    PutOnFloor,
    MoveToOrigin,
    RotateToZ,
    KeypointsToTomato,
    RandomCropKeypoints,
    MaskedCrop,
    MaskedRandomCrop
)

__all__ = [
    'Compose', 'to_tensor', 'Transpose', 'Collect', 'WrapFieldsToLists',
    'ToTensor', 'Crop', 'RandomCrop', 'Normalize', 'SwapSiameseMotion',
    'ProcessSiameseMotion', 'LoadMotion', 'RetargetSkeleton', 'MotionDownsample',
    'PutOnFloor', 'MoveToOrigin', 'RotateToZ', 'KeypointsToTomato', 'RandomCropKeypoints',
    'MaskedCrop', 'MaskedRandomCrop'
]