From 68f906cc67c688e0f2921b465aacdc113eea9a6b Mon Sep 17 00:00:00 2001 From: AlbertvanHouten Date: Fri, 18 Mar 2022 10:16:07 +0100 Subject: [PATCH] Added Rotated detection --- ote_sdk/ote_sdk/entities/model_template.py | 1 + ote_sdk/ote_sdk/tests/entities/test_model_template.py | 1 + 2 files changed, 2 insertions(+) diff --git a/ote_sdk/ote_sdk/entities/model_template.py b/ote_sdk/ote_sdk/entities/model_template.py index 8a77ceb0354..191d4c1cf0e 100644 --- a/ote_sdk/ote_sdk/entities/model_template.py +++ b/ote_sdk/ote_sdk/entities/model_template.py @@ -457,6 +457,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 index d01e6045869..3c61bebc395 100644 --- a/ote_sdk/ote_sdk/tests/entities/test_model_template.py +++ b/ote_sdk/ote_sdk/tests/entities/test_model_template.py @@ -1040,6 +1040,7 @@ def test_task_type_constants(self): TaskType.ANOMALY_DETECTION, TaskType.ANOMALY_CLASSIFICATION, TaskType.ANOMALY_SEGMENTATION, + TaskType.ROTATED_DETECTION, )