sapiens/dense/src/datasets/transforms/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from .albedo_transforms import (
    AlbedoPackInputs,
    AlbedoRandomCrop,
    AlbedoRandomCropContinuous,
    AlbedoRandomFlip,
    AlbedoRandomScale,
    AlbedoResize,
    AlbedoResizePadImage,
)
from .normal_transforms import (
    NormalGenerateTarget,
    NormalPackInputs,
    NormalRandomCrop,
    NormalRandomCropContinuous,
    NormalRandomFlip,
    NormalRandomScale,
    NormalResize,
    NormalResizePadImage,
)
from .pointmap_transforms import (
    PointmapGenerateTarget,
    PointmapPackInputs,
    PointmapRandomCrop,
    PointmapRandomCropContinuous,
    PointmapRandomFlip,
    PointmapRandomScale,
    PointmapResize,
    PointmapResizePadImage,
)
from .seg_transforms import (
    SegPackInputs,
    SegRandomBackground,
    SegRandomCrop,
    SegRandomHorizontalFlip,
    SegRandomResize,
    SegRandomRotate,
    SegResize,
)

__all__ = [
    "SegRandomBackground",
    "SegRandomResize",
    "SegPackInputs",
    "SegRandomCrop",
    "SegRandomHorizontalFlip",
    "SegRandomRotate",
    "SegResize",
    "PointmapGenerateTarget",
    "PointmapPackInputs",
    "PointmapRandomCrop",
    "PointmapRandomCropContinuous",
    "PointmapRandomFlip",
    "PointmapRandomScale",
    "PointmapResize",
    "PointmapResizePadImage",
    "NormalGenerateTarget",
    "NormalPackInputs",
    "NormalRandomCrop",
    "NormalRandomCropContinuous",
    "NormalRandomFlip",
    "NormalRandomScale",
    "NormalResize",
    "NormalResizePadImage",
    "AlbedoPackInputs",
    "AlbedoRandomCrop",
    "AlbedoRandomCropContinuous",
    "AlbedoRandomFlip",
    "AlbedoRandomScale",
    "AlbedoResize",
    "AlbedoResizePadImage",
]