mindspore.ops.IOU
- class mindspore.ops.IOU(mode='iou')[source]
Calculates intersection over union for boxes.
Computes the intersection over union (IOU) or the intersection over foreground (IOF) based on the ground-truth and predicted regions.
Refer to
mindspore.ops.iou()
for more detail.- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> iou = ops.IOU(mode='iou') >>> anchor_boxes = Tensor(np.random.randint(1.0, 5.0, [3, 4]), mindspore.float16) >>> gt_boxes = Tensor(np.random.randint(1.0, 5.0, [3, 4]), mindspore.float16) >>> output = iou(anchor_boxes, gt_boxes) >>> print(output.shape) (3, 3)