showing annotated images for detect target message bug fix

modules/detect_target/detect_target_factory.py

@@ -27,8 +27,12 @@ def create_detect_target(
     save_name: str,
 ) -> tuple[bool, base_detect_target.BaseDetectTarget | None]:
     """
-    Construct detect target class at runtime.
+    Factory function to create a detection target object.
+
+    Returns:
+    Tuple containing success status and the instantiated detection object (if successful).
     """
+
     match detect_target_option:
         case DetectTargetOption.ML_ULTRALYTICS:
             return True, detect_target_ultralytics.DetectTargetUltralytics(

modules/detect_target/detect_target_ultralytics.py

@@ -3,9 +3,9 @@
 """
 
 import time
-
 import cv2
 import ultralytics
+import torch
 
 from . import base_detect_target
 from .. import image_and_time
@@ -37,7 +37,7 @@ def __init__(
         self.__device = device
         self.__model = ultralytics.YOLO(model_path)
         self.__counter = 0
-        self.__enable_half_precision = not self.__device == "cpu"
+        self.__enable_half_precision = self.__device != "cpu"
         self.__local_logger = local_logger
         self.__show_annotations = show_annotations
         if override_full:
@@ -58,6 +58,10 @@ def run(
         """
         image = data.image
         start_time = time.time()
+        # Fall back to CPU if no GPU is available
+        if self.__device != "cpu" and not torch.cuda.is_available():
+            self.__local_logger.warning("CUDA not available. Falling back to CPU.")
+            self.__device = "cpu"
 
         predictions = self.__model.predict(
             source=image,
@@ -111,6 +115,11 @@ def run(
             self.__counter += 1
 
         if self.__show_annotations:
-            cv2.imshow("Annotated", image_annotated)  # type: ignore
+            if image_annotated is not None:
+                # Display the annotated image in a named window
+                cv2.imshow("Annotated", image_annotated)
+                cv2.waitKey(1)  # Short delay to process GUI events
+            else:
+                self.__local_logger.warning("Annotated image is invalid.")
 
         return True, detections