dgs.models.embedding_generator.torchreid.TorchreidEmbeddingGenerator.predict_ids

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

Predict class IDs given some input.

Parameters:

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

Returns:

Tensor containing class predictions, which are not necessarily a probability distribution. Shape: [B x num_classes]