CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-06 12:43:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

guard for tobytes()

Browse Source
This commit is contained in:
henryruhs
2026-06-05 21:39:30 +02:00
parent fd3e8b82e2
commit 0cbfa5a415
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/face_store.py
+8 -4
View File
@@ -1,5 +1,7 @@
from typing import List, Optional
import numpy
from facefusion.hash_helper import create_hash
from facefusion.types import Face, FaceStore, VisionFrame
@@ -7,13 +9,15 @@ FACE_STORE : FaceStore = {}
def get_faces(vision_frame : VisionFrame) -> Optional[List[Face]]:
vision_hash = create_hash(vision_frame.tobytes())
return FACE_STORE.get(vision_hash)
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:
vision_hash = create_hash(vision_frame.tobytes())
if vision_hash:
if numpy.any(vision_frame):
vision_hash = create_hash(vision_frame.tobytes())
FACE_STORE[vision_hash] = faces