This commit is contained in:
galthran-wq 2025-02-11 13:27:13 +00:00
parent 0ef420bc10
commit 72919d95f4

View File

@ -150,7 +150,7 @@ def represent(
batch_confidences.append(confidence) batch_confidences.append(confidence)
# Convert list of images to a numpy array for batch processing # Convert list of images to a numpy array for batch processing
batch_images = np.concat(batch_images) batch_images = np.concatenate(batch_images, axis=0)
# 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)