administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-08 06:25:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Cleanup bird classification

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-14 07:05:24 -06:00
parent 759d963a04
commit 70697a93b4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/data_processing/real_time/bird.py
View File

@ -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(