administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-08-02 17:12:16 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fixes (#20102)

Browse Source 
* Catch bird classification resize error

* Improve openvino width detection

* Use auto by default

* Set type
This commit is contained in:
Nicolas Mowen
2025-09-16 16:06:51 -06:00
committed by GitHub
parent 975c8485f9
commit 26178444f3
3 changed files with 37 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frigate/data_processing/real_time/bird.py
+5 -1
View File
@@ -128,7 +128,11 @@ class BirdRealTimeProcessor(RealTimeProcessorApi):
]
if input.shape != (224, 224):
input = cv2.resize(input, (224, 224))
try:
input = cv2.resize(input, (224, 224))
except Exception:
logger.warning("Failed to resize image for bird classification")
return
input = np.expand_dims(input, axis=0)
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input)
frigate/data_processing/real_time/custom_classification.py
+12 -4
View File
@@ -133,8 +133,12 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
x:x2,
]
if frame.shape != (224, 224):
frame = cv2.resize(frame, (224, 224))
if input.shape != (224, 224):
try:
input = cv2.resize(input, (224, 224))
except Exception:
logger.warning("Failed to resize image for state classification")
return
input = np.expand_dims(frame, axis=0)
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input)
@@ -254,8 +258,12 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
x:x2,
]
if crop.shape != (224, 224):
crop = cv2.resize(crop, (224, 224))
if input.shape != (224, 224):
try:
input = cv2.resize(input, (224, 224))
except Exception:
logger.warning("Failed to resize image for object classification")
return
input = np.expand_dims(crop, axis=0)
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input)