diff --git a/deepface/modules/representation.py b/deepface/modules/representation.py index a017114..6bdbf1b 100644 --- a/deepface/modules/representation.py +++ b/deepface/modules/representation.py @@ -154,6 +154,8 @@ def represent( # Forward pass through the model for the entire batch embeddings = model.forward(batch_images) + if len(batch_images) == 1: + embeddings = [embeddings] for embedding, region, confidence in zip(embeddings, batch_regions, batch_confidences): resp_objs.append(