administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-16 03:52:09 +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,
]
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)