Catch bird classification resize error

This commit is contained in:
Nicolas Mowen 2025-09-16 14:55:46 -06:00
parent 975c8485f9
commit c0c9099616

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)