kaolin.metrics.voxelgrid¶
API¶
-
kaolin.metrics.voxelgrid.
iou
(pred, gt)¶ Computes IoU across two voxelgrids
Parameters: - pred (torch.Tensor) – predicted (binary) voxelgrids of shape (N, X, Y, Z)
- gt (torch.Tensor) – ground-truth (binary) voxelgrids of shape (N, X, Y, Z)
Returns: the intersection over union value
Return type: (torch.FloatTensor)
Example
>>> pred = torch.tensor([[[[0., 0.], ... [1., 1.]], ... [[1., 1.], ... [1., 1.]]]]) >>> gt = torch.ones((1,2,2,2)) >>> iou(pred, gt) tensor([0.7500])