CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-27 04:21:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

margin decouple (#1199)

Browse Source
This commit is contained in:
Harisreedhar
2026-07-24 21:19:35 +05:30
committed by GitHub
parent c5f892e76c
commit e25efafef0
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/video_manager.py
+2 -1
View File
@@ -69,6 +69,7 @@ def read_video_reader_window(video_reader : VideoReader, frame_start : int, fram
id = video_reader.get('id')
frame_set = frame_store.get_frame_store(id)
buffer_margin = 16
keep_margin = 4
frame_gaps = []
for frame_index in range(frame_start, frame_end + 1):
@@ -88,7 +89,7 @@ def read_video_reader_window(video_reader : VideoReader, frame_start : int, fram
if numpy.any(vision_frame):
frame_store.set_frame(id, decode_index, vision_frame)
frame_store.reduce_frames(id, frame_start - buffer_margin, frame_end + buffer_margin)
frame_store.reduce_frames(id, frame_start - keep_margin, frame_end + keep_margin)
return frame_store.select_frame_set(id, frame_start, frame_end)
tests/test_video_manager.py
+1 -1
View File
@@ -121,7 +121,7 @@ def test_read_video_reader_window() -> None:
read_video_reader_window(video_reader, 21, 25)
assert min(get_frame_store(video_reader.get('id'))) == 5
assert min(get_frame_store(video_reader.get('id'))) == 17
assert max(get_frame_store(video_reader.get('id'))) == 25
frame_set = read_video_reader_window(video_reader, 268, 275)