osc.viz.loss_objects.viz_contrastive_loss_objects_probs

viz_contrastive_loss_objects_probs(obj_feats, temp, loss)

Visualize contrastive loss between 2B images with S objects each, probabilities only.

Parameters

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

• temp (float) – loss temperature

• loss (float) – loss value to display on the plots

Return type

Figure

Returns

A figure with a single plot of matching probabilities.