CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-09-20 14:32:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

port 3.8.3 correct

Browse Source
This commit is contained in:
henryruhs
2026-09-01 16:19:17 +02:00
parent 8e6178269b
commit 366153941a
16 changed files with 229 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_vision.py
+17 -1
View File
@@ -4,7 +4,7 @@ import pytest
from facefusion import ffmpeg, ffmpeg_builder, process_manager
from facefusion.download import conditional_download
from facefusion.vision import calculate_histogram_difference, count_video_frame_total, detect_image_resolution, detect_video_duration, detect_video_fps, detect_video_resolution, match_frame_color, normalize_resolution, pack_resolution, predict_video_frame_total, read_image, read_video_frame, restrict_image_resolution, restrict_trim_video_frame, restrict_video_fps, restrict_video_resolution, scale_resolution, select_video_frames, unpack_resolution, write_image
from facefusion.vision import calculate_histogram_difference, count_video_frame_total, detect_image_resolution, detect_video_duration, detect_video_fps, detect_video_resolution, match_frame_color, normalize_resolution, pack_resolution, predict_video_frame_total, read_image, read_video_frame, resolve_extract_frame_index, resolve_target_frame_index, restrict_image_resolution, restrict_trim_video_frame, restrict_video_fps, restrict_video_resolution, scale_resolution, select_video_frames, unpack_resolution, write_image
from .assert_helper import get_test_example_file, get_test_examples_directory, get_test_output_path, prepare_test_output_directory
@@ -154,9 +154,25 @@ def test_predict_video_frame_total() -> None:
assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 12.5, 0, 100) == 50
assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 25, 0, 100) == 100
assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 25, 0, 200) == 200
assert predict_video_frame_total(get_test_example_file('target-240p-25fps.mp4'), 6.25, 0, 270) == 68
assert predict_video_frame_total(get_test_example_file('target-240p-30fps.mp4'), 9.49, 40, 200) == 50
assert predict_video_frame_total('invalid', 25, 0, 100) == 0
def test_resolve_extract_frame_index() -> None:
assert resolve_extract_frame_index(25, 12.5, 100) == 50
assert resolve_extract_frame_index(25, 25, 100) == 100
assert resolve_extract_frame_index(25, 6.25, 270) == 68
assert resolve_extract_frame_index(30, 9.49, 40) == 13
def test_resolve_target_frame_index() -> None:
assert resolve_target_frame_index(25, 12.5, 50) == 100
assert resolve_target_frame_index(25, 25, 100) == 100
assert resolve_target_frame_index(25, 6.25, 1) == 5
assert resolve_target_frame_index(30, 9.49, 13) == 42
def test_detect_video_fps() -> None:
assert detect_video_fps(get_test_example_file('target-240p-25fps.mp4')) == 25.0
assert detect_video_fps(get_test_example_file('target-240p-30fps.mp4')) == 30.0