Merge branch 'add_in_classical_cv' of https://github.com/UWARG/comput…

…er-vision-python into add_in_classical_cv
UWARG · Nov 28, 2024 · e68e659 · e68e659
2 parents de3e2e4 + 15793cd
commit e68e659
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/config.yaml b/config.yaml
@@ -11,7 +11,7 @@ detect_target:
     worker_count: 1
     option: 0  # 0 is for Ultralytics (from detect_target_factory.py)
     device: 0
-    model_path: "tests/model_example/yolov8s_ultralytics_pretrained_default.pt"  # TODO: update
+    model_path: "tests/model_example/best-2s.pt"  # TODO: update
     save_prefix: "log_comp"
 
 flight_interface:

diff --git a/modules/detect_target/detect_target_factory.py b/modules/detect_target/detect_target_factory.py
@@ -51,5 +51,10 @@ def create_detect_target(
                 show_annotations,
                 save_name,
             )
+        case DetectTargetOption.C_CONTOUR:
+            return True, detect_target_contour.DetectTargetContour(
+                show_annotations,
+                save_name,
+            )
 
     return False, None