dgs.models.similarity.torchreid.TorchreidVisualSimilarity.get_data

TorchreidVisualSimilarity.get_data(ds: State) → torch.Tensor

Given a State get the current embedding or compute it using the image crop.