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

Correctly check for objects

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-23 06:09:43 -06:00
parent e7427b47fb
commit 60212a9aa4
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/data_processing/real_time/teachable_machine.py
View File

@ -127,7 +127,7 @@ class TeachableMachineObjectProcessor(RealTimeProcessorApi):
self.labelmap = load_labels(self.model_config.labelmap_path, prefill=0)
def process_frame(self, obj_data, frame):
if obj_data["label"] != "object":
if obj_data["label"] not in self.model_config.object_config.objects:
return
x, y, x2, y2 = calculate_region(

4
frigate/embeddings/maintainer.py
View File

@ -503,9 +503,7 @@ class EmbeddingMaintainer(threading.Thread):
if isinstance(processor, LicensePlateRealTimeProcessor):
processor.process_frame(camera, yuv_frame, True)
if isinstance(processor, TeachableMachineObjectProcessor) or isinstance(
processor, TeachableMachineStateProcessor
):
if isinstance(processor, TeachableMachineStateProcessor):
processor.process_frame({"camera": camera}, yuv_frame)
self.frame_manager.close(frame_name)