github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 11:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

compatibility

Browse Source
This commit is contained in:
galthran-wq 2025-02-11 13:56:09 +00:00
parent d7a985bf12
commit bb134b25d2
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/modules/representation.py
View File

@ -154,6 +154,8 @@ def represent(
# Forward pass through the model for the entire batch # Forward pass through the model for the entire batch
embeddings = model.forward(batch_images) embeddings = model.forward(batch_images)
if len(batch_images) == 1:
embeddings = [embeddings]
for embedding, region, confidence in zip(embeddings, batch_regions, batch_confidences): for embedding, region, confidence in zip(embeddings, batch_regions, batch_confidences):
resp_objs.append( resp_objs.append(