kaolin.metrics.render

API

kaolin.metrics.render.mask_iou(lhs_mask, rhs_mask)

Compute the Intersection over Union of two segmentation masks.

Parameters
  • lhs_mask (torch.FloatTensor) – A segmentation mask, of shape \((\text{batch_size}, \text{height}, \text{width})\).

  • rhs_mask (torch.FloatTensor) – A segmentation mask, of shape \((\text{batch_size}, \text{height}, \text{width})\).

Returns

The IoU loss, as a torch scalar.

Return type

(torch.FloatTensor)