Add rotated det OV recipe (#3687)

* Add rotated det ov recipe * fix e2e test cli
openvinotoolkit · Jun 26, 2024 · 8cc9542 · 8cc9542
1 parent ead8a02
commit 8cc9542
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/src/otx/recipe/rotated_detection/openvino_model.yaml b/src/otx/recipe/rotated_detection/openvino_model.yaml
@@ -0,0 +1,25 @@
+model:
+  class_path: otx.core.model.instance_segmentation.OVInstanceSegmentationModel
+  init_args:
+    label_info: 80
+    model_name: openvino.xml
+    model_type: MaskRCNN
+    async_inference: True
+    use_throughput_mode: True
+
+engine:
+  task: ROTATED_DETECTION
+  device: cpu
+
+callback_monitor: val/map_50
+
+data: ../_base_/data/torchvision_base.yaml
+overrides:
+  data:
+    task: ROTATED_DETECTION
+    config:
+      include_polygons: true
+      image_color_channel: RGB
+      data_format: coco_instances
+      test_subset:
+        batch_size: 24
diff --git a/tests/e2e/cli/test_cli.py b/tests/e2e/cli/test_cli.py
@@ -125,7 +125,6 @@ def test_otx_e2e_cli(
         task_name in recipe
         for task_name in [
             "dino_v2",
-            "rotated_detection/maskrcnn",
         ]
     ):
         return

diff --git a/tests/integration/cli/test_cli.py b/tests/integration/cli/test_cli.py
@@ -148,7 +148,6 @@ def test_otx_e2e(
         task_name in recipe
         for task_name in [
             "dino_v2",
-            "rotated_detection/maskrcnn",
         ]
     ):
         return