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

Save original frame

Browse Source
This commit is contained in:
Nicolas Mowen 2025-10-07 09:19:58 -06:00
parent 03cbe650f8
commit 5b8ec7b8b7
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -142,7 +142,7 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
if frame.shape != (224, 224): if frame.shape != (224, 224):
try: try:
frame = cv2.resize(frame, (224, 224)) resized_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
@ -158,7 +158,7 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
) )
return return
input = np.expand_dims(frame, axis=0) input = np.expand_dims(resized_frame, axis=0)
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input) self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input)
self.interpreter.invoke() self.interpreter.invoke()
res: np.ndarray = self.interpreter.get_tensor( res: np.ndarray = self.interpreter.get_tensor(
@ -286,7 +286,7 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
if crop.shape != (224, 224): if crop.shape != (224, 224):
try: try:
crop = cv2.resize(crop, (224, 224)) resized_crop = cv2.resize(crop, (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
@ -302,7 +302,7 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
) )
return return
input = np.expand_dims(crop, axis=0) input = np.expand_dims(resized_crop, axis=0)
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input) self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input)
self.interpreter.invoke() self.interpreter.invoke()
res: np.ndarray = self.interpreter.get_tensor( res: np.ndarray = self.interpreter.get_tensor(