From e25efafef0c9dd2b1a5e3249fb5b5934b695aec2 Mon Sep 17 00:00:00 2001 From: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> Date: Fri, 24 Jul 2026 21:19:35 +0530 Subject: [PATCH] margin decouple (#1199) --- facefusion/video_manager.py | 3 ++- tests/test_video_manager.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/facefusion/video_manager.py b/facefusion/video_manager.py index fcfbbae4..a6716786 100644 --- a/facefusion/video_manager.py +++ b/facefusion/video_manager.py @@ -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) diff --git a/tests/test_video_manager.py b/tests/test_video_manager.py index d6e4e55c..89dfdb15 100644 --- a/tests/test_video_manager.py +++ b/tests/test_video_manager.py @@ -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)