dgs.models.embedding_generator.torchreid.TorchreidEmbeddingGenerator.predict_embeddings

TorchreidEmbeddingGenerator.predict_embeddings(data: torch.Tensor) → torch.Tensor

Predict embeddings given some input.

Parameters:

data – The input for the model, most likely a cropped image.

Returns:

Tensor containing a batch B of embeddings. Shape: [B x E]