diff --git a/src/otx/recipe/rotated_detection/openvino_model.yaml b/src/otx/recipe/rotated_detection/openvino_model.yaml new file mode 100644 index 00000000000..10179896456 --- /dev/null +++ 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 index cc7c6544efd..68f72b9baa7 100644 --- 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 index 2a03a588f4a..435c763a600 100644 --- 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