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

Adjust input format

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-07 05:37:19 -06:00
parent ce2e5031b8
commit 584d131a7f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/data_processing/post/review_descriptions.py
View File

@ -20,7 +20,7 @@ class ReviewDescriptionProcessor(PostProcessorApi):
self.tracked_review_items: dict[str, list[tuple[int, bytes]]] = {} self.tracked_review_items: dict[str, list[tuple[int, bytes]]] = {}
self.genai_client = OllamaClient( self.genai_client = OllamaClient(
GenAIConfig( GenAIConfig(
enabled=True, model="gemma3:4b", base_url="http://192.168.50.107:11434" enabled=True, model="qwen2.5vl:3b", base_url="http://192.168.50.107:11434"
) )
) )
@ -62,7 +62,8 @@ class ReviewDescriptionProcessor(PostProcessorApi):
self.genai_client.generate_review_description( self.genai_client.generate_review_description(
{ {
"camera": final_data["camera"], "camera": final_data["camera"],
"objects": final_data["data"]["objects"] + final_data["data"]["sub_labels"], "objects": final_data["data"]["objects"],
"recognized_objects": final_data["data"]["sub_labels"],
"zones": final_data["data"]["zones"], "zones": final_data["data"]["zones"],
}, },
[r[1] for r in self.tracked_review_items[id]], [r[1] for r in self.tracked_review_items[id]],