engine/data/transforms/__init__.py

"""
Copied from RT-DETR (https://github.com/lyuwenyu/RT-DETR)
Copyright(c) 2023 lyuwenyu. All Rights Reserved.
"""


from ._transforms import (
    EmptyTransform,
    RandomPhotometricDistort,
    RandomZoomOut,
    RandomIoUCrop,
    RandomHorizontalFlip,
    Resize,
    PadToSize,
    SanitizeBoundingBoxes,
    RandomCrop,
    Normalize,
    ConvertBoxes,
    ConvertPILImage,
)
from .container import Compose
from .mosaic import Mosaic