osc.viz.loss_objects.viz_contrastive_loss_objects

viz_contrastive_loss_objects(obj_feats, loss)

Visualize contrastive loss 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.

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

Return type

Figure

Returns

A figure with two plots.