dgs.models.dgs.dgs.DGSModule.forward

DGSModule.forward(ds: State, target: State, **kwargs) → torch.Tensor

Given a State containing the current detections and a target, compute the similarity between every pair.

Returns:

The combined similarity matrix as tensor of shape [nof_detections x (nof_tracks + nof_detections)].