administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-09 15:05:26 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

chore: ruff

Browse Source
This commit is contained in:
ZhaiSoul 2026-04-14 16:33:46 +08:00
parent 1577657a6d
commit 36a08a9af5
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
frigate/data_processing/real_time/custom_classification.py
View File

@ -434,9 +434,13 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
return return
input = np.expand_dims(crop, axis=0) input = np.expand_dims(crop, 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 = self.interpreter.get_tensor(self.tensor_output_details[0]["index"])[0] res = 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 = int(np.argmax(probs)) best_id = int(np.argmax(probs))
@ -472,14 +476,18 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
if object_id not in self.classification_history: if object_id not in self.classification_history:
self.classification_history[object_id] = [] self.classification_history[object_id] = []
self.classification_history[object_id].append((label, score, timestamp)) self.classification_history[object_id].append(
(label, score, timestamp)
)
logger.debug( logger.debug(
f"Added classification result for {object_id}: {label} ({score}) at {timestamp}" f"Added classification result for {object_id}: {label} ({score}) at {timestamp}"
) )
self._publish_result(object_id, camera, label, score, timestamp) self._publish_result(object_id, camera, label, score, timestamp)
except Exception as e: except Exception as e:
logger.error(f"Error in inference thread for {object_id}: {e}", exc_info=True) logger.error(
f"Error in inference thread for {object_id}: {e}", exc_info=True
)
finally: finally:
with self.interpreter_lock: with self.interpreter_lock:
self.active_tasks[object_id] = self.active_tasks.get(object_id, 1) - 1 self.active_tasks[object_id] = self.active_tasks.get(object_id, 1) - 1
@ -661,7 +669,9 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
with self.interpreter_lock: with self.interpreter_lock:
current_tasks = self.active_tasks.get(object_id, 0) current_tasks = self.active_tasks.get(object_id, 0)
if current_tasks >= MAX_CONCURRENT_CLASSIFICATIONS_PER_OBJECT: if current_tasks >= MAX_CONCURRENT_CLASSIFICATIONS_PER_OBJECT:
logger.debug(f"Object {object_id} has {current_tasks}/{MAX_CONCURRENT_CLASSIFICATIONS_PER_OBJECT} active tasks") logger.debug(
f"Object {object_id} has {current_tasks}/{MAX_CONCURRENT_CLASSIFICATIONS_PER_OBJECT} active tasks"
)
return return
now = datetime.datetime.now().timestamp() now = datetime.datetime.now().timestamp()