showing annotated images for detect target message bug fix

modules/detect_target/detect_target_factory.py

@@ -27,7 +27,10 @@ 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:
+    Success, detect target object.
     """
     match detect_target_option:
         case DetectTargetOption.ML_ULTRALYTICS:
@@ -39,5 +42,4 @@ def create_detect_target(
                 show_annotations,
                 save_name,
             )
-
     return False, None

modules/detect_target/detect_target_ultralytics.py

@@ -5,6 +5,7 @@
 import time
 
 import cv2
+import torch
 import ultralytics
 
 from . import base_detect_target
@@ -37,14 +38,17 @@ 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:
             self.__enable_half_precision = False
         self.__filename_prefix = ""
         if save_name != "":
             self.__filename_prefix = save_name + "_" + str(int(time.time())) + "_"
+        if self.__device != "cpu" and not torch.cuda.is_available():
+            self.__local_logger.warning("CUDA not available. Falling back to CPU.")
+            self.__device = "cpu"
 
     def run(
         self, data: image_and_time.ImageAndTime
@@ -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