Fix classification cropping

This commit is contained in:
Nicolas Mowen 2025-09-18 10:09:12 -06:00
parent 620db568b0
commit 323c237c6d

View File

@ -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