Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-20 13:54:36 +01:00
Code Issues Projects Releases Wiki Activity

Cleanup bird classification (#17146)

Browse Source 
* Cleanup bird classification

* Cleanup
This commit is contained in:
Nicolas Mowen
2025-03-14 07:21:50 -06:00
committed by GitHub
parent 759d963a04
commit fbd22c8124
2 changed files with 6 additions and 6 deletions
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,
]
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)
self.interpreter.set_tensor(self.tensor_input_details[0]["index"], input)
self.interpreter.invoke()
res: np.ndarray = self.interpreter.get_tensor(