1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
- # PACKAGE IMPORTS FOR EXTERNAL USAGE
- import cv2
- from super_gradients.training.transforms.transforms import (
- DetectionTransform,
- DetectionStandardize,
- DetectionMosaic,
- DetectionRandomAffine,
- DetectionHSV,
- DetectionRGB2BGR,
- DetectionPaddedRescale,
- DetectionTargetsFormatTransform,
- Standardize,
- )
- from super_gradients.common.object_names import Transforms
- from super_gradients.common.registry.registry import TRANSFORMS
- from super_gradients.common.registry.albumentation import ALBUMENTATIONS_TRANSFORMS, ALBUMENTATIONS_COMP_TRANSFORMS, imported_albumentations_failure
- __all__ = [
- "TRANSFORMS",
- "ALBUMENTATIONS_TRANSFORMS",
- "ALBUMENTATIONS_COMP_TRANSFORMS",
- "imported_albumentations_failure",
- "Transforms",
- "DetectionTransform",
- "DetectionStandardize",
- "DetectionMosaic",
- "DetectionRandomAffine",
- "DetectionHSV",
- "DetectionRGB2BGR",
- "DetectionPaddedRescale",
- "DetectionTargetsFormatTransform",
- "Standardize",
- ]
- cv2.setNumThreads(0)
|