mirror of
https://github.com/facefusion/facefusion.git
synced 2026-06-10 22:37:48 +02:00
Henry Ruhs
dbf4346f7b
* bring back the face store but for source and reference only * fix ci * minor improvement * guard for tobytes()
26 lines
620 B
Python
26 lines
620 B
Python
from typing import List, Optional
|
|
|
|
import numpy
|
|
|
|
from facefusion.hash_helper import create_hash
|
|
from facefusion.types import Face, FaceStore, VisionFrame
|
|
|
|
FACE_STORE : FaceStore = {}
|
|
|
|
|
|
def get_faces(vision_frame : VisionFrame) -> Optional[List[Face]]:
|
|
if numpy.any(vision_frame):
|
|
vision_hash = create_hash(vision_frame.tobytes())
|
|
return FACE_STORE.get(vision_hash)
|
|
return None
|
|
|
|
|
|
def set_faces(vision_frame : VisionFrame, faces : List[Face]) -> None:
|
|
if numpy.any(vision_frame):
|
|
vision_hash = create_hash(vision_frame.tobytes())
|
|
FACE_STORE[vision_hash] = faces
|
|
|
|
|
|
def clear_faces() -> None:
|
|
FACE_STORE.clear()
|