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 (B, H, W)
rhs_mask (torch.FloatTensor) – A segmentation mask of shape (B, H, W)
- Returns
The IoU loss, as a torch scalar.
- Return type
(torch.FloatTensor)