fix data pipeline

openvinotoolkit · Jun 17, 2024 · d64175f · d64175f
1 parent b3b13ba
commit d64175f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/.../algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/data_pipeline.py b/.../algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/data_pipeline.py
@@ -24,7 +24,7 @@
         resize_cfg=dict(
             type="Resize",
             img_scale=__img_size,
-            keep_ratio=True,
+            keep_ratio=False,
         ),
         enable_memcache=True,  # Cache after resizing image & annotations
     ),
@@ -54,7 +54,7 @@
 val_pipeline = [
     dict(
         type="LoadResizeDataFromOTXDataset",
-        resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=True),
+        resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=False),
         enable_memcache=True,  # Cache after resizing image
     ),
     dict(

diff --git a/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/data_pipeline.py b/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/data_pipeline.py
@@ -22,7 +22,7 @@
         resize_cfg=dict(
             type="Resize",
             img_scale=__img_size,
-            keep_ratio=True,
+            keep_ratio=False,
         ),
         enable_memcache=True,  # Cache after resizing image & annotations
     ),
@@ -52,7 +52,7 @@
 val_pipeline = [
     dict(
         type="LoadResizeDataFromOTXDataset",
-        resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=True),
+        resize_cfg=dict(type="Resize", img_scale=__img_size, keep_ratio=False),
         enable_memcache=True,  # Cache after resizing image
     ),
     dict(
@@ -76,7 +76,7 @@
         img_scale=__img_size,
         flip=False,
         transforms=[
-            dict(type="Resize", keep_ratio=True),
+            dict(type="Resize", keep_ratio=False),
             dict(type="RandomFlip"),
             dict(type="Normalize", **__img_norm_cfg),
             dict(type="Pad", size_divisor=32),

diff --git a/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/deployment.py b/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/deployment.py
@@ -8,5 +8,5 @@
 
 backend_config = dict(
     # dynamic batch causes forever running openvino process
-    model_inputs=[dict(opt_shapes=dict(input=[1, 3, 800, 1344]))],
+    model_inputs=[dict(opt_shapes=dict(input=[1, 3, 1024, 1024]))],
 )