diff --git a/facefusion/face_store.py b/facefusion/face_store.py index 0356378e..8e627b0a 100644 --- a/facefusion/face_store.py +++ b/facefusion/face_store.py @@ -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