super-gradients/src/super_gradients/training/transforms/__init__.py

# PACKAGE IMPORTS FOR EXTERNAL USAGE
import cv2
from super_gradients.training.transforms.transforms import (
    DetectionMosaic,
    DetectionRandomAffine,
    DetectionHSV,
    DetectionPaddedRescale,
    DetectionTargetsFormatTransform,
)
from super_gradients.training.transforms.all_transforms import (
    TRANSFORMS,
    ALBUMENTATIONS_TRANSFORMS,
    Transforms,
    ALBUMENTATIONS_COMP_TRANSFORMS,
    imported_albumentations_failure,
)

__all__ = [
    "TRANSFORMS",
    "ALBUMENTATIONS_TRANSFORMS",
    "ALBUMENTATIONS_COMP_TRANSFORMS",
    "Transforms",
    "DetectionMosaic",
    "DetectionRandomAffine",
    "DetectionHSV",
    "DetectionPaddedRescale",
    "DetectionTargetsFormatTransform",
    "imported_albumentations_failure",
]

cv2.setNumThreads(0)