Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#561 Feature/sg 193 extend output formator

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-193-extend_detection_target_transform
@@ -1,5 +1,5 @@
 from super_gradients.common.factories.base_factory import BaseFactory
 from super_gradients.common.factories.base_factory import BaseFactory
-from super_gradients.training.utils.bbox_formats import BBOX_FORMATS
+from super_gradients.training.datasets.data_formats.bbox_formats import BBOX_FORMATS
 
 
 
 
 class BBoxFormatFactory(BaseFactory):
 class BBoxFormatFactory(BaseFactory):
Discard
1
2
3
4
5
6
7
  1. from super_gradients.common.factories.type_factory import TypeFactory
  2. from super_gradients.training.datasets.data_formats.default_formats import DEFAULT_CONCATENATED_TENSOR_FORMATS
  3. class ConcatenatedTensorFormatFactory(TypeFactory):
  4. def __init__(self):
  5. super().__init__(DEFAULT_CONCATENATED_TENSOR_FORMATS)
Discard
@@ -50,7 +50,6 @@ class Transforms:
     DetectionHSV = "DetectionHSV"
     DetectionHSV = "DetectionHSV"
     DetectionHorizontalFlip = "DetectionHorizontalFlip"
     DetectionHorizontalFlip = "DetectionHorizontalFlip"
     DetectionPaddedRescale = "DetectionPaddedRescale"
     DetectionPaddedRescale = "DetectionPaddedRescale"
-    DetectionTargetsFormat = "DetectionTargetsFormat"
     DetectionTargetsFormatTransform = "DetectionTargetsFormatTransform"
     DetectionTargetsFormatTransform = "DetectionTargetsFormatTransform"
     RandomResizedCropAndInterpolation = "RandomResizedCropAndInterpolation"
     RandomResizedCropAndInterpolation = "RandomResizedCropAndInterpolation"
     RandAugmentTransform = "RandAugmentTransform"
     RandAugmentTransform = "RandAugmentTransform"
@@ -253,3 +252,18 @@ class Models:
     PP_LITE_B_SEG75 = "pp_lite_b_seg75"
     PP_LITE_B_SEG75 = "pp_lite_b_seg75"
     UNET_CUSTOM = "unet_custom"
     UNET_CUSTOM = "unet_custom"
     UNET_CUSTOM_CLS = "unet_custom_cls"
     UNET_CUSTOM_CLS = "unet_custom_cls"
+
+
+class ConcatenatedTensorFormats:
+    XYXY_LABEL = "XYXY_LABEL"
+    XYWH_LABEL = "XYWH_LABEL"
+    CXCYWH_LABEL = "CXCYWH_LABEL"
+    LABEL_XYXY = "LABEL_XYXY"
+    LABEL_XYWH = "LABEL_XYWH"
+    LABEL_CXCYWH = "LABEL_CXCYWH"
+    NORMALIZED_XYXY_LABEL = "NORMALIZED_XYXY_LABEL"
+    NORMALIZED_XYWH_LABEL = "NORMALIZED_XYWH_LABEL"
+    NORMALIZED_CXCYWH_LABEL = "NORMALIZED_CXCYWH_LABEL"
+    LABEL_NORMALIZED_XYXY = "LABEL_NORMALIZED_XYXY"
+    LABEL_NORMALIZED_XYWH = "LABEL_NORMALIZED_XYWH"
+    LABEL_NORMALIZED_CXCYWH = "LABEL_NORMALIZED_CXCYWH"
Discard