Fix KP Export (single obj case) (#4191)

remove transform for single object case
openvinotoolkit · Jan 29, 2025 · 35fdb7f · 35fdb7f
1 parent 40b82f6
commit 35fdb7f
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/otx/recipe/keypoint_detection/openvino_model.yaml b/src/otx/recipe/keypoint_detection/openvino_model.yaml
@@ -26,7 +26,6 @@ overrides:
       batch_size: 1
       num_workers: 2
       transforms:
-        - class_path: otx.core.data.transform_libs.torchvision.GetBBoxCenterScale
         - class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
           init_args:
             input_size: $(input_size)
@@ -35,7 +34,6 @@ overrides:
       batch_size: 1
       num_workers: 2
       transforms:
-        - class_path: otx.core.data.transform_libs.torchvision.GetBBoxCenterScale
         - class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
           init_args:
             input_size: $(input_size)
@@ -44,7 +42,6 @@ overrides:
       batch_size: 64
       num_workers: 2
       transforms:
-        - class_path: otx.core.data.transform_libs.torchvision.GetBBoxCenterScale
         - class_path: otx.core.data.transform_libs.torchvision.TopdownAffine
           init_args:
             input_size: $(input_size)