mindspore.ops.IOU
- class mindspore.ops.IOU(mode='iou')[源代码]
计算矩形的IOU,即真实区域和预测区域的交并比。
根据真实区域和预测区域计算IOU(intersection over union)或IOF(intersection over foreground)。
更多参考详见
mindspore.ops.iou()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> 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)