66ea4928f8
* Allow passing the onnxruntime to install.py * Fix CI * Disallow none execution providers in the UI * Use CV2 to detect fps * Respect trim on videos with audio * Respect trim on videos with audio (finally) * Implement caching to speed up preview and webcam * Define webcam UI and webcam performance * Remove layout from components * Add primary buttons * Extract benchmark and webcam settings * Introduce compact UI settings * Caching for IO and **** prediction * Caching for IO and **** prediction * Introduce face analyser caching * Fix some typing * Improve setup for benchmark * Clear image cache via post process * Fix settings in UI, Simplify restore_audio() using shortest * Select resolution and fps via webcam ui * Introduce read_static_image() to stop caching temp images * Use DirectShow under Windows * Multi-threading for webcam * Fix typing * Refactor frame processor * Refactor webcam processing * Avoid warnings due capture.isOpened() * Resume downloads (#110) * Introduce resumable downloads * Fix CURL commands * Break execution_settings into pieces * Cosmetic changes on cv2 webcam * Update Gradio * Move face cache to own file * Uniform namings for threading * Fix sorting of get_temp_frame_paths(), extend get_temp_frames_pattern() * Minor changes from the review * Looks stable to tme * Update the disclaimer * Update the disclaimer * Update the disclaimer
41 lines
1.7 KiB
Python
41 lines
1.7 KiB
Python
from typing import List, Optional
|
|
import gradio
|
|
|
|
import facefusion.globals
|
|
from facefusion import wording
|
|
from facefusion.processors.frame.core import load_frame_processor_module, clear_frame_processors_modules
|
|
from facefusion.uis import core as ui
|
|
from facefusion.uis.typing import Update
|
|
from facefusion.utilities import list_module_names
|
|
|
|
FRAME_PROCESSORS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
|
|
|
|
|
|
def render() -> None:
|
|
global FRAME_PROCESSORS_CHECKBOX_GROUP
|
|
|
|
FRAME_PROCESSORS_CHECKBOX_GROUP = gradio.CheckboxGroup(
|
|
label = wording.get('frame_processors_checkbox_group_label'),
|
|
choices = sort_frame_processors(facefusion.globals.frame_processors),
|
|
value = facefusion.globals.frame_processors
|
|
)
|
|
ui.register_component('frame_processors_checkbox_group', FRAME_PROCESSORS_CHECKBOX_GROUP)
|
|
|
|
|
|
def listen() -> None:
|
|
FRAME_PROCESSORS_CHECKBOX_GROUP.change(update_frame_processors, inputs = FRAME_PROCESSORS_CHECKBOX_GROUP, outputs = FRAME_PROCESSORS_CHECKBOX_GROUP)
|
|
|
|
|
|
def update_frame_processors(frame_processors : List[str]) -> Update:
|
|
clear_frame_processors_modules()
|
|
facefusion.globals.frame_processors = frame_processors
|
|
for frame_processor in facefusion.globals.frame_processors:
|
|
frame_processor_module = load_frame_processor_module(frame_processor)
|
|
frame_processor_module.pre_check()
|
|
return gradio.update(value = frame_processors, choices = sort_frame_processors(frame_processors))
|
|
|
|
|
|
def sort_frame_processors(frame_processors : List[str]) -> list[str]:
|
|
frame_processors_names = list_module_names('facefusion/processors/frame/modules')
|
|
return sorted(frame_processors_names, key = lambda frame_processor : frame_processors.index(frame_processor) if frame_processor in frame_processors else len(frame_processors))
|