From f87e82481d188399dba7a9b3ec84cb8d69144a95 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sat, 11 Jan 2025 16:15:41 -0700 Subject: [PATCH] Get classification working --- frigate/data_processing/real_time/bird_processor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/frigate/data_processing/real_time/bird_processor.py b/frigate/data_processing/real_time/bird_processor.py index c96fb1868..cf2c5bfea 100644 --- a/frigate/data_processing/real_time/bird_processor.py +++ b/frigate/data_processing/real_time/bird_processor.py @@ -93,7 +93,6 @@ class BirdProcessor(RealTimeProcessorApi): x:x2, ] - logger.info(f"input shape is {input.shape}") cv2.imwrite("/media/frigate/test_class.png", input) input = np.expand_dims(input, axis=0) @@ -103,7 +102,8 @@ class BirdProcessor(RealTimeProcessorApi): res: np.ndarray = self.interpreter.get_tensor(self.tensor_output_details[0]["index"])[0] probs = res / res.sum(axis=0) best_id = np.argmax(probs) - score = probs[best_id] + score = round(probs[best_id], 2) + logger.info(f"the best scoring index is {best_id} {score}%") def handle_request(self, request_data): return None