From fbd22c8124d4980025ccc6d3a408731d3bfca7af Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 14 Mar 2025 07:21:50 -0600 Subject: [PATCH] Cleanup bird classification (#17146) * Cleanup bird classification * Cleanup --- frigate/api/media.py | 8 ++++---- frigate/data_processing/real_time/bird.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/frigate/api/media.py b/frigate/api/media.py index 3c7fc09da..83307a15c 100644 --- a/frigate/api/media.py +++ b/frigate/api/media.py @@ -781,14 +781,14 @@ def event_snapshot( height=params.height, quality=params.quality, ) - except Exception as e: + except Exception: return JSONResponse( - content={"success": False, "message": f"Ongoing event not found: {e}"}, + content={"success": False, "message": "Ongoing event not found"}, status_code=404, ) - except Exception as e: + except Exception: return JSONResponse( - content={"success": False, "message": f"Unknown error occurred: {e}"}, + content={"success": False, "message": "Unknown error occurred"}, status_code=404, ) diff --git a/frigate/data_processing/real_time/bird.py b/frigate/data_processing/real_time/bird.py index d942edf6f..ba6d4f08c 100644 --- a/frigate/data_processing/real_time/bird.py +++ b/frigate/data_processing/real_time/bird.py @@ -115,10 +115,10 @@ class BirdRealTimeProcessor(RealTimeProcessorApi): x:x2, ] - cv2.imwrite("/media/frigate/test_class.png", input) + if input.shape != (224, 224): + input = cv2.resize(input, (224, 224)) input = np.expand_dims(input, axis=0) - self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input) self.interpreter.invoke() res: np.ndarray = self.interpreter.get_tensor(