osc.viz.loss_objects.match_objects

match_objects(obj_feats)

Compute matches between 2B images with S objects each.

Parameters

obj_feats (Tensor) – object feature tensor of shape [2B S C], where S is the number of objects per image.

Return type

ndarray

Returns

An array of matches (one match per row), length 2BS