From 620db568b0a302d78d34bdd9d198d606f0656b82 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 18 Sep 2025 10:06:22 -0600 Subject: [PATCH] Fix --- .../data_processing/real_time/custom_classification.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index daa9fee96..19cdc4ece 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -133,9 +133,9 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi): x:x2, ] - if input.shape != (224, 224): + if frame.shape != (224, 224): try: - input = cv2.resize(input, (224, 224)) + frame = cv2.resize(frame, (224, 224)) except Exception: logger.warning("Failed to resize image for state classification") return @@ -258,11 +258,11 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi): x:x2, ] - if input.shape != (224, 224): + if frame.shape != (224, 224): try: - input = cv2.resize(input, (224, 224)) + frame = cv2.resize(frame, (224, 224)) except Exception: - logger.warning("Failed to resize image for object classification") + logger.warning("Failed to resize image for state classification") return input = np.expand_dims(crop, axis=0)