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
30 lines
687 B
Python
30 lines
687 B
Python
from typing import Optional, List, Dict
|
|
import hashlib
|
|
|
|
from facefusion.typing import Frame, Face
|
|
|
|
FACES_CACHE : Dict[str, List[Face]] = {}
|
|
|
|
|
|
def get_faces_cache(frame : Frame) -> Optional[List[Face]]:
|
|
frame_hash = create_frame_hash(frame)
|
|
if frame_hash in FACES_CACHE:
|
|
return FACES_CACHE[frame_hash]
|
|
return None
|
|
|
|
|
|
def set_faces_cache(frame : Frame, faces : List[Face]) -> None:
|
|
frame_hash = create_frame_hash(frame)
|
|
if frame_hash:
|
|
FACES_CACHE[frame_hash] = faces
|
|
|
|
|
|
def clear_faces_cache() -> None:
|
|
global FACES_CACHE
|
|
|
|
FACES_CACHE = {}
|
|
|
|
|
|
def create_frame_hash(frame : Frame) -> Optional[str]:
|
|
return hashlib.sha256(frame.tobytes()).hexdigest() if frame is not None else None
|