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

add area to face and license_plate attributes in debug and snapshots

Browse Source
This commit is contained in:
Josh Hawkins 2025-05-17 07:25:59 -05:00
parent ebae6cb1ed
commit 8057106b71
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/camera/state.py
View File

@ -172,6 +172,7 @@ class CameraState:
# draw any attributes
for attribute in obj["current_attributes"]:
box = attribute["box"]
box_area = int((box[2] - box[0]) * (box[3] - box[1]))
draw_box_with_label(
frame_copy,
box[0],
@ -179,7 +180,7 @@ class CameraState:
box[2],
box[3],
attribute["label"],
f"{attribute['score']:.0%}",
f"{attribute['score']:.0%}{' ' + str(box_area) if attribute['label'] in ['license_plate', 'face'] else ''}",
thickness=thickness,
color=color,
)

3
frigate/track/tracked_object.py
View File

@ -474,6 +474,7 @@ class TrackedObject:
# draw any attributes
for attribute in self.thumbnail_data["attributes"]:
box = attribute["box"]
box_area = int((box[2] - box[0]) * (box[3] - box[1]))
draw_box_with_label(
best_frame,
box[0],
@ -481,7 +482,7 @@ class TrackedObject:
box[2],
box[3],
attribute["label"],
f"{attribute['score']:.0%}",
f"{attribute['score']:.0%}{' ' + str(box_area) if attribute['label'] in ['license_plate', 'face'] else ''}",
thickness=thickness,
color=color,
)