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

Get classification working

Browse Source
This commit is contained in:
Nicolas Mowen 2025-01-11 16:15:41 -07:00
parent f86b2232df
commit f87e82481d
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_processor.py
View File

@ -93,7 +93,6 @@ class BirdProcessor(RealTimeProcessorApi):
x:x2, x:x2,
] ]
logger.info(f"input shape is {input.shape}")
cv2.imwrite("/media/frigate/test_class.png", input) cv2.imwrite("/media/frigate/test_class.png", input)
input = np.expand_dims(input, axis=0) input = np.expand_dims(input, axis=0)
@ -103,7 +102,8 @@ class BirdProcessor(RealTimeProcessorApi):
res: np.ndarray = self.interpreter.get_tensor(self.tensor_output_details[0]["index"])[0] res: np.ndarray = self.interpreter.get_tensor(self.tensor_output_details[0]["index"])[0]
probs = res / res.sum(axis=0) probs = res / res.sum(axis=0)
best_id = np.argmax(probs) best_id = np.argmax(probs)
score = probs[best_id] score = round(probs[best_id], 2)
logger.info(f"the best scoring index is {best_id} {score}%")
def handle_request(self, request_data): def handle_request(self, request_data):
return None return None