Formatting

Add debug logs
2026-03-22 08:08:22 +03:00 · 2025-11-09 07:27:33 -07:00 · 2025-11-09 07:16:56 -07:00
1 changed files with 6 additions and 0 deletions
--- a/frigate/data_processing/real_time/custom_classification.py
+++ b/frigate/data_processing/real_time/custom_classification.py
@ -227,6 +227,9 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
            self.tensor_output_details[0]["index"]
        )[0]
        probs = res / res.sum(axis=0)
+        logger.debug(
+            f"{self.model_config.name} Ran state classification with probabilities: {probs}"
+        )
        best_id = np.argmax(probs)
        score = round(probs[best_id], 2)
        self.__update_metrics(datetime.datetime.now().timestamp() - now)
@ -455,6 +458,9 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
            self.tensor_output_details[0]["index"]
        )[0]
        probs = res / res.sum(axis=0)
+        logger.debug(
+            f"{self.model_config.name} Ran object classification with probabilities: {probs}"
+        )
        best_id = np.argmax(probs)
        score = round(probs[best_id], 2)
        self.__update_metrics(datetime.datetime.now().timestamp() - now)
Author	SHA1	Message	Date
Nicolas Mowen	a242c31b9e	Formatting	2025-11-09 07:27:33 -07:00
Nicolas Mowen	8700332a44	Add debug logs	2025-11-09 07:16:56 -07:00