From 0ef5de1d02453620ac396d93ebb5429d0f8f3b49 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 5 Jun 2026 21:39:30 +0200 Subject: [PATCH] guard for tobytes() (cherry picked from commit 0cbfa5a415e55559e8b2aeb41ae688eb2f180c5a) --- facefusion/face_store.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) 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