Merge pull request #984 from openvinotoolkit/albert/CVS-81984

Added Rotated detection
openvinotoolkit · Mar 18, 2022 · 4d1ae2d · 4d1ae2d
2 parents abafaa3 + 68f906c
commit 4d1ae2d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/ote_sdk/ote_sdk/entities/model_template.py b/ote_sdk/ote_sdk/entities/model_template.py
@@ -477,6 +477,7 @@ def __init__(self) -> None:
     TaskType.ANOMALY_DETECTION,
     TaskType.ANOMALY_CLASSIFICATION,
     TaskType.ANOMALY_SEGMENTATION,
+    TaskType.ROTATED_DETECTION,
 )
 
 

diff --git a/ote_sdk/ote_sdk/tests/entities/test_model_template.py b/ote_sdk/ote_sdk/tests/entities/test_model_template.py
@@ -1049,6 +1049,7 @@ def test_task_type_constants(self):
             TaskType.ANOMALY_DETECTION,
             TaskType.ANOMALY_CLASSIFICATION,
             TaskType.ANOMALY_SEGMENTATION,
+            TaskType.ROTATED_DETECTION,
         )