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 = []