mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-07-30 13:48:07 +02:00
Cleanup bird classification (#17146)
* Cleanup bird classification * Cleanup
This commit is contained in:
parent
759d963a04
commit
fbd22c8124
@ -781,14 +781,14 @@ def event_snapshot(
|
|||||||
height=params.height,
|
height=params.height,
|
||||||
quality=params.quality,
|
quality=params.quality,
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception:
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
content={"success": False, "message": f"Ongoing event not found: {e}"},
|
content={"success": False, "message": "Ongoing event not found"},
|
||||||
status_code=404,
|
status_code=404,
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception:
|
||||||
return JSONResponse(
|
return JSONResponse(
|
||||||
content={"success": False, "message": f"Unknown error occurred: {e}"},
|
content={"success": False, "message": "Unknown error occurred"},
|
||||||
status_code=404,
|
status_code=404,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -115,10 +115,10 @@ class BirdRealTimeProcessor(RealTimeProcessorApi):
|
|||||||
x:x2,
|
x:x2,
|
||||||
]
|
]
|
||||||
|
|
||||||
cv2.imwrite("/media/frigate/test_class.png", input)
|
if input.shape != (224, 224):
|
||||||
|
input = cv2.resize(input, (224, 224))
|
||||||
|
|
||||||
input = np.expand_dims(input, axis=0)
|
input = np.expand_dims(input, 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(
|
||||||
|
Loading…
Reference in New Issue
Block a user