diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index 19cdc4ece..841267a60 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -258,9 +258,9 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi): x:x2, ] - if frame.shape != (224, 224): + if crop.shape != (224, 224): try: - frame = cv2.resize(frame, (224, 224)) + crop = cv2.resize(crop, (224, 224)) except Exception: logger.warning("Failed to resize image for state classification") return