From a242c31b9e12875b262211ef982dcc2ed75a4bf6 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 9 Nov 2025 07:27:33 -0700 Subject: [PATCH] Formatting --- .../data_processing/real_time/custom_classification.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index b2235b1e7..45ec30338 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -227,7 +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}") + 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) @@ -456,7 +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}") + 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)