jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Vibe coded benchmark command part3

Browse Source
This commit is contained in:
henryruhs
2025-06-15 20:23:52 +02:00
parent 703c64bed5
commit 45fbac0220
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
facefusion/types.py
View File

@@ -100,6 +100,16 @@ LogLevelSet : TypeAlias = Dict[LogLevel, int]
TableHeaders = List[str]
TableContents = List[List[Any]]
BenchmarkSet = TypedDict('BenchmarkSet',
{
'target_path' : str,
'benchmark_cycles' : int,
'average_run' : float,
'fastest_run' : float,
'slowest_run' : float,
'relative_fps' : float
})
FaceDetectorModel = Literal['many', 'retinaface', 'scrfd', 'yolo_face']
FaceLandmarkerModel = Literal['many', '2dfan4', 'peppa_wutz']
FaceDetectorSet : TypeAlias = Dict[FaceDetectorModel, List[str]]

3
facefusion/uis/components/benchmark.py
View File

@@ -56,4 +56,5 @@ def start(benchmark_resolutions : List[str], benchmark_cycles : int) -> Generato
state_manager.sync_item('execution_thread_count')
state_manager.sync_item('execution_queue_count')
yield from benchmarker.run()
for benchmarks in benchmarker.run_with_progress():
yield [ list(benchmark.values()) for benchmark in benchmarks ]