administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-30 18:47:40 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Don't set sub label for none object classification

Browse Source
This commit is contained in:
Nicolas Mowen 2025-06-26 18:40:41 -06:00
parent 9d7f5e5438
commit 93fd0fa775
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -150,11 +150,14 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
score, score,
) )
if score >= camera_config.threshold: if score >= camera_config.threshold :
self.requestor.send_data( sub_label = self.labelmap[best_id]
f"{camera}/classification/{self.model_config.name}",
self.labelmap[best_id], if sub_label.lower() != "none":
) self.requestor.send_data(
f"{camera}/classification/{self.model_config.name}",
self.labelmap[best_id],
)
def handle_request(self, topic, request_data): def handle_request(self, topic, request_data):
if topic == EmbeddingsRequestEnum.reload_classification_model.value: if topic == EmbeddingsRequestEnum.reload_classification_model.value: