Add additional transformations to D-Fine configuration in YAML files

openvinotoolkit · Jan 7, 2025 · ad9169e · ad9169e
1 parent 86bf6a8
commit ad9169e
Show file tree

Hide file tree

Showing 4 changed files with 64 additions and 0 deletions.
diff --git a/src/otx/recipe/detection/dfine_l.yaml b/src/otx/recipe/detection/dfine_l.yaml
@@ -74,10 +74,26 @@ overrides:
             transform_bbox: true
             keep_ratio: false
             is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.PhotoMetricDistortion
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.RandomAffine
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: torchvision.transforms.v2.GaussianBlur
+          enable: false
+          init_args:
+            kernel_size: 5
+        - class_path: torchvision.transforms.v2.RandomVerticalFlip
+          enable: false
         - class_path: torchvision.transforms.v2.ToDtype
           init_args:
             dtype: ${as_torch_dtype:torch.float32}
             scale: true
+        - class_path: torchvision.transforms.v2.GaussianNoise
+          enable: false
         - class_path: torchvision.transforms.v2.SanitizeBoundingBoxes
           init_args:
             min_size: 1

diff --git a/src/otx/recipe/detection/dfine_l_tile.yaml b/src/otx/recipe/detection/dfine_l_tile.yaml
@@ -85,10 +85,26 @@ overrides:
             transform_bbox: true
             keep_ratio: false
             is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.PhotoMetricDistortion
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.RandomAffine
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: torchvision.transforms.v2.GaussianBlur
+          enable: false
+          init_args:
+            kernel_size: 5
+        - class_path: torchvision.transforms.v2.RandomVerticalFlip
+          enable: false
         - class_path: torchvision.transforms.v2.ToDtype
           init_args:
             dtype: ${as_torch_dtype:torch.float32}
             scale: true
+        - class_path: torchvision.transforms.v2.GaussianNoise
+          enable: false
         - class_path: torchvision.transforms.v2.SanitizeBoundingBoxes
           init_args:
             min_size: 1

diff --git a/src/otx/recipe/detection/dfine_x.yaml b/src/otx/recipe/detection/dfine_x.yaml
@@ -74,10 +74,26 @@ overrides:
             transform_bbox: true
             keep_ratio: false
             is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.PhotoMetricDistortion
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.RandomAffine
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: torchvision.transforms.v2.GaussianBlur
+          enable: false
+          init_args:
+            kernel_size: 5
+        - class_path: torchvision.transforms.v2.RandomVerticalFlip
+          enable: false
         - class_path: torchvision.transforms.v2.ToDtype
           init_args:
             dtype: ${as_torch_dtype:torch.float32}
             scale: true
+        - class_path: torchvision.transforms.v2.GaussianNoise
+          enable: false
         - class_path: torchvision.transforms.v2.SanitizeBoundingBoxes
           init_args:
             min_size: 1

diff --git a/src/otx/recipe/detection/dfine_x_tile.yaml b/src/otx/recipe/detection/dfine_x_tile.yaml
@@ -85,10 +85,26 @@ overrides:
             transform_bbox: true
             keep_ratio: false
             is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.PhotoMetricDistortion
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: otx.core.data.transform_libs.torchvision.RandomAffine
+          enable: false
+          init_args:
+            is_numpy_to_tvtensor: true
+        - class_path: torchvision.transforms.v2.GaussianBlur
+          enable: false
+          init_args:
+            kernel_size: 5
+        - class_path: torchvision.transforms.v2.RandomVerticalFlip
+          enable: false
         - class_path: torchvision.transforms.v2.ToDtype
           init_args:
             dtype: ${as_torch_dtype:torch.float32}
             scale: true
+        - class_path: torchvision.transforms.v2.GaussianNoise
+          enable: false
         - class_path: torchvision.transforms.v2.SanitizeBoundingBoxes
           init_args:
             min_size: 1