D-FINE/src/data/transforms/__init__.py

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

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