Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-03-04 00:17:22 +01:00
Code Issues Projects Releases Wiki Activity

Ensure label does not overlap with box or go out of frame (#15376)

Browse Source
This commit is contained in:
Josh Hawkins 2024-12-06 09:32:16 -06:00 committed by GitHub
parent d3b631a952
commit 8aa6297308
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
frigate/util/image.py
View File

@ -219,6 +219,8 @@ def draw_box_with_label(
text_width = size[0][0] text_width = size[0][0]
text_height = size[0][1] text_height = size[0][1]
line_height = text_height + size[1] line_height = text_height + size[1]
# get frame height
frame_height = frame.shape[0]
# set the text start position # set the text start position
if position == "ul": if position == "ul":
text_offset_x = x_min text_offset_x = x_min
@ -228,18 +230,23 @@ def draw_box_with_label(
text_offset_y = max(0, y_min - (line_height + 8)) text_offset_y = max(0, y_min - (line_height + 8))
elif position == "bl": elif position == "bl":
text_offset_x = x_min text_offset_x = x_min
text_offset_y = y_max text_offset_y = min(frame_height - line_height, y_max)
elif position == "br": elif position == "br":
text_offset_x = max(0, x_max - (text_width + 8)) text_offset_x = max(0, x_max - (text_width + 8))
text_offset_y = y_max text_offset_y = min(frame_height - line_height, y_max)
# Adjust position if it overlaps with the box or goes out of frame
# Adjust position if it overlaps with the box if position in {"ul", "ur"}:
if position in {"ul", "ur"} and text_offset_y < y_min + thickness: if text_offset_y < y_min + thickness: # Label overlaps with the box
# Move the text below the box if y_min - (line_height + 8) < 0 and y_max + line_height <= frame_height:
text_offset_y = y_max # Not enough space above, and there is space below
elif position in {"bl", "br"} and text_offset_y + line_height > y_max: text_offset_y = y_max
# Move the text above the box elif y_min - (line_height + 8) >= 0:
text_offset_y = max(0, y_min - (line_height + 8)) # Enough space above, keep the label at the top
text_offset_y = max(0, y_min - (line_height + 8))
elif position in {"bl", "br"}:
if text_offset_y + line_height > frame_height:
# If there's not enough space below, try above the box
text_offset_y = max(0, y_min - (line_height + 8))
# make the coords of the box with a small padding of two pixels # make the coords of the box with a small padding of two pixels
textbox_coords = ( textbox_coords = (