administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-24 15:47:43 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-18 10:06:22 -06:00
parent 85bff61776
commit 620db568b0
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/data_processing/real_time/custom_classification.py
View File

@ -133,9 +133,9 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
x:x2, x:x2,
] ]
if input.shape != (224, 224): if frame.shape != (224, 224):
try: try:
input = cv2.resize(input, (224, 224)) frame = cv2.resize(frame, (224, 224))
except Exception: except Exception:
logger.warning("Failed to resize image for state classification") logger.warning("Failed to resize image for state classification")
return return
@ -258,11 +258,11 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
x:x2, x:x2,
] ]
if input.shape != (224, 224): if frame.shape != (224, 224):
try: try:
input = cv2.resize(input, (224, 224)) frame = cv2.resize(frame, (224, 224))
except Exception: except Exception:
logger.warning("Failed to resize image for object classification") logger.warning("Failed to resize image for state classification")
return return
input = np.expand_dims(crop, axis=0) input = np.expand_dims(crop, axis=0)