diff --git a/facefusion/vision.py b/facefusion/vision.py index 72b211f..d03e383 100644 --- a/facefusion/vision.py +++ b/facefusion/vision.py @@ -104,8 +104,8 @@ def count_video_frame_total(video_path : str) -> int: def predict_video_frame_total(video_path : str, fps : Fps, trim_frame_start : int, trim_frame_end : int) -> int: if is_video(video_path): - target_video_fps = detect_video_fps(video_path) - extract_frame_total = count_trim_frame_total(video_path, trim_frame_start, trim_frame_end) * fps / target_video_fps + video_fps = detect_video_fps(video_path) + extract_frame_total = count_trim_frame_total(video_path, trim_frame_start, trim_frame_end) * fps / video_fps return math.floor(extract_frame_total) return 0