dgs.models.metric.metric.TorchreidCosineDistance

class dgs.models.metric.metric.TorchreidCosineDistance(*args: Any, **kwargs: Any)

Call TorchReid’s version of the cosine distance.

Methods

__init__(*args: Any, **kwargs: Any) → None

static forward(input1: torch.Tensor, input2: torch.Tensor) → torch.Tensor