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

Add appropriate fields to data

Browse Source
This commit is contained in:
Nick Mowen 2023-04-29 12:39:01 -06:00
parent 5372d4668e
commit f295305a06
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
frigate/events.py
View File

@ -141,17 +141,21 @@ class EventProcessor(threading.Thread):
Event.camera: camera, Event.camera: camera,
Event.start_time: start_time, Event.start_time: start_time,
Event.end_time: end_time, Event.end_time: end_time,
Event.top_score: event_data["top_score"],
Event.score: score,
Event.zones: list(event_data["entered_zones"]), Event.zones: list(event_data["entered_zones"]),
Event.thumbnail: event_data["thumbnail"], Event.thumbnail: event_data["thumbnail"],
Event.region: region,
Event.box: box,
Event.has_clip: event_data["has_clip"], Event.has_clip: event_data["has_clip"],
Event.has_snapshot: event_data["has_snapshot"], Event.has_snapshot: event_data["has_snapshot"],
Event.model_hash: first_detector.model.model_hash, Event.model_hash: first_detector.model.model_hash,
Event.model_type: first_detector.model.model_type, Event.model_type: first_detector.model.model_type,
Event.detector_type: first_detector.type, Event.detector_type: first_detector.type,
Event.data: {
"area": event_data.get("area", 0),
"box": box,
"ratio": event_data.get("ratio", width / height),
"region": region,
"score": score,
"top_score": event_data["top_score"],
}
} }
( (

2
frigate/models.py
View File

@ -44,7 +44,7 @@ class Event(Model): # type: ignore[misc]
model_hash = CharField(max_length=32) model_hash = CharField(max_length=32)
detector_type = CharField(max_length=32) detector_type = CharField(max_length=32)
model_type = CharField(max_length=32) model_type = CharField(max_length=32)
data = JSONField() # ex: tracked object area, box, region, etc. data = JSONField() # ex: tracked object box, region, etc.
class Timeline(Model): # type: ignore[misc] class Timeline(Model): # type: ignore[misc]