Add base engine recipe and update recipes for det

openvinotoolkit · Jul 8, 2024 · 22d648b · 22d648b
1 parent 502163b
commit 22d648b
Show file tree

Hide file tree

Showing 16 changed files with 23 additions and 90 deletions.
diff --git a/src/otx/recipe/_base_/engine/detection.yaml b/src/otx/recipe/_base_/engine/detection.yaml
@@ -0,0 +1,5 @@
+engine:
+  task: DETECTION
+  device: auto
+
+callback_monitor: val/map_50
diff --git a/src/otx/recipe/detection/atss_mobilenetv2.yaml b/src/otx/recipe/detection/atss_mobilenetv2.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/atss_mobilenetv2_tile.yaml b/src/otx/recipe/detection/atss_mobilenetv2_tile.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/atss_resnext101.yaml b/src/otx/recipe/detection/atss_resnext101.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/openvino_model.yaml b/src/otx/recipe/detection/openvino_model.yaml
@@ -7,14 +7,12 @@ model:
     model_type: "SSD"
     async_inference: True
 
-engine:
-  task: DETECTION
-  device: cpu
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:
+  engine:
+    device: cpu
+
   data:
     stack_images: false
     test_subset:

diff --git a/src/otx/recipe/detection/rtmdet_tiny.yaml b/src/otx/recipe/detection/rtmdet_tiny.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/ssd_mobilenetv2.yaml b/src/otx/recipe/detection/ssd_mobilenetv2.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/ssd_mobilenetv2_tile.yaml b/src/otx/recipe/detection/ssd_mobilenetv2_tile.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_l.yaml b/src/otx/recipe/detection/yolox_l.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_l_tile.yaml b/src/otx/recipe/detection/yolox_l_tile.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_s.yaml b/src/otx/recipe/detection/yolox_s.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_s_tile.yaml b/src/otx/recipe/detection/yolox_s_tile.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_tiny.yaml b/src/otx/recipe/detection/yolox_tiny.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_tiny_tile.yaml b/src/otx/recipe/detection/yolox_tiny_tile.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_x.yaml b/src/otx/recipe/detection/yolox_x.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides:

diff --git a/src/otx/recipe/detection/yolox_x_tile.yaml b/src/otx/recipe/detection/yolox_x_tile.yaml
@@ -1,9 +1,4 @@
-engine:
-  task: DETECTION
-  device: auto
-
-callback_monitor: val/map_50
-
+engine: ../_base_/engine/detection.yaml
 model: ../_base_/model/detection.yaml
 data: ../_base_/data/detection.yaml
 overrides: