From 8adca77ca64984b89f3c72d8a046b77a2ed27cc6 Mon Sep 17 00:00:00 2001 From: kremnik Date: Mon, 30 Sep 2024 19:55:13 +0300 Subject: [PATCH] fix: variable name has been changed --- deepface/modules/recognition.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/deepface/modules/recognition.py b/deepface/modules/recognition.py index 12a368c..a4848eb 100644 --- a/deepface/modules/recognition.py +++ b/deepface/modules/recognition.py @@ -505,7 +505,7 @@ def find_batched( """ embeddings_list = [] valid_mask = [] - other_keys = set() + metadata = set() for item in representations: emb = item.get('embedding') @@ -516,18 +516,18 @@ def find_batched( embeddings_list.append(np.zeros_like(representations[0]['embedding'])) valid_mask.append(False) - other_keys.update(item.keys()) + metadata.update(item.keys()) # remove embedding key from other keys - other_keys.discard('embedding') - other_keys = list(other_keys) + metadata.discard('embedding') + metadata = list(metadata) embeddings = np.array(embeddings_list) # (N, D) valid_mask = np.array(valid_mask) # (N,) data = { key: np.array([item.get(key, None) for item in representations]) - for key in other_keys + for key in metadata } target_embeddings = []