Added logs to detect target

modules/detect_target/detect_target_factory.py

@@ -6,6 +6,7 @@
 
 from . import base_detect_target
 from . import detect_target_ultralytics
+from ..common.logger.modules import logger
 
 
 class DetectTargetOption(enum.Enum):
@@ -23,6 +24,7 @@ def create_detect_target(
     override_full: bool,
     show_annotations: bool,
     save_name: str,
+    local_logger: logger.Logger,
 ) -> tuple[bool, base_detect_target.BaseDetectTarget | None]:
     """
     Construct detect target class at runtime.
@@ -33,6 +35,7 @@ def create_detect_target(
                 device,
                 model_path,
                 override_full,
+                local_logger,
                 show_annotations,
                 save_name,
             )

modules/detect_target/detect_target_ultralytics.py

@@ -10,6 +10,7 @@
 from . import base_detect_target
 from .. import image_and_time
 from .. import detections_and_time
+from ..common.logger.modules import logger
 
 
 class DetectTargetUltralytics(base_detect_target.BaseDetectTarget):
@@ -22,6 +23,7 @@ def __init__(
         device: "str | int",
         model_path: str,
         override_full: bool,
+        local_logger: logger.Logger,
         show_annotations: bool = False,
         save_name: str = "",
     ) -> None:
@@ -37,6 +39,8 @@ def __init__(
         self.__counter = 0
         self.__enable_half_precision = not self.__device == "cpu"
         self.__show_annotations = show_annotations
+        self.__logger = local_logger
+
         if override_full:
             self.__enable_half_precision = False
         self.__filename_prefix = ""
@@ -54,6 +58,7 @@ def run(
         Return: Success and the detections.
         """
         image = data.image
+        start_time = time.time()
         predictions = self.__model.predict(
             source=image,
             half=self.__enable_half_precision,
@@ -108,4 +113,8 @@ def run(
         if self.__show_annotations:
             cv2.imshow("Annotated", image_annotated)  # type: ignore
 
+        end_time = time.time()
+        self.__logger.info(
+            f"{time.localtime()}: Target detection took {round(end_time - start_time, 3)} seconds"
+        )
         return True, detections

modules/detect_target/detect_target_worker.py

@@ -49,6 +49,7 @@ def detect_target_worker(
         override_full,
         show_annotations,
         save_name,
+        local_logger,
     )
 
     if not result:

tests/unit/test_detect_target_ultralytics.py

@@ -13,6 +13,7 @@
 from modules.detect_target import detect_target_ultralytics
 from modules import image_and_time
 from modules import detections_and_time
+from modules.common.logger.modules import logger
 
 
 TEST_PATH = pathlib.Path("tests", "model_example")
@@ -108,8 +109,13 @@ def detector() -> detect_target_ultralytics.DetectTargetUltralytics:  # type: ig
     """
     Construct DetectTargetUltralytics.
     """
+
+    _, test_logger = logger.Logger.create("test_logger", False)
+
+    assert test_logger is not None
+
     detection = detect_target_ultralytics.DetectTargetUltralytics(
-        DEVICE, str(MODEL_PATH), OVERRIDE_FULL
+        DEVICE, str(MODEL_PATH), OVERRIDE_FULL, test_logger
     )
     yield detection  # type: ignore