nik/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-03-04 06:33:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adjust frame selection logic

Browse Source
This commit is contained in:
Nicolas Mowen 2025-08-15 06:17:40 -06:00
parent 9d2f5e5db5
commit 9db1ccb3f9
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -169,10 +169,8 @@ class ReviewDescriptionProcessor(PostProcessorApi):
) -> list[str]:
preview_dir = os.path.join(CACHE_DIR, "preview_frames")
file_start = f"preview_{camera}"
# 30 seconds padding to ensure that a frame before & after the activity is considered
start_file = f"{file_start}-{start_time - 30}.webp"
end_file = f"{file_start}-{end_time + 30}.webp"
start_file = f"{file_start}-{start_time}.webp"
end_file = f"{file_start}-{end_time}.webp"
all_frames = []
for file in sorted(os.listdir(preview_dir)):
@ -180,9 +178,15 @@ class ReviewDescriptionProcessor(PostProcessorApi):
continue
if file < start_file:
if len(all_frames):
all_frames[0] = os.path.join(preview_dir, file)
else:
all_frames.append(os.path.join(preview_dir, file))
continue
if file > end_file:
all_frames.append(os.path.join(preview_dir, file))
break
all_frames.append(os.path.join(preview_dir, file))
@ -217,7 +221,9 @@ def run_analysis(
"id": final_data["id"],
"camera": camera,
"zones": final_data["data"]["zones"],
"start": datetime.datetime.fromtimestamp(final_data["start_time"]).strftime("%A, %I:%M %p"),
"start": datetime.datetime.fromtimestamp(final_data["start_time"]).strftime(
"%A, %I:%M %p"
),
"duration": final_data["end_time"] - final_data["start_time"],
}