forked from huawei/mindspore2022
33 lines
1.3 KiB
ReStructuredText
33 lines
1.3 KiB
ReStructuredText
mindspore.ops.IOU
|
||
=================
|
||
|
||
.. py:class:: mindspore.ops.IOU(mode='iou')
|
||
|
||
计算矩形的IOU，即真实区域和预测区域的交并比。
|
||
|
||
根据真实区域和预测区域计算IOU(intersection over union)或IOF(intersection over foreground)。
|
||
|
||
.. math::
|
||
\text{IOU} = \frac{\text{Area of Overlap}}{\text{Area of Union}}
|
||
|
||
\text{IOF} = \frac{\text{Area of Overlap}}{\text{Area of Ground Truth}}
|
||
|
||
.. warning::
|
||
在Ascend中，仅支持计算float16数据。为避免溢出，输入长度和宽度在内部按0.2缩放。
|
||
|
||
**参数：**
|
||
|
||
- **mode** (string) - 指定计算方法，现支持'iou'(intersection over union)或'iof'(intersection over foreground)模式。默认值：'iou'。
|
||
|
||
**输入：**
|
||
|
||
- **anchor_boxes** (Tensor) - 预测区域，shape为(N, 4)的Tensor。"N"表示预测区域的数量，"4"表示"x0"、"y0"、"x1"和"y1"。数据类型为float16或float32。
|
||
- **gt_boxes** (Tensor) - 真实区域，shape为(M, 4)的Tensor。"M"表示地面真实区域的数量，"4"表示"x0"、"y0"、"x1"和"y1"。数据类型为float16或float32。
|
||
|
||
**输出：**
|
||
|
||
IOU值的Tensor，shape为(M, N)的Tensor，数据类型与 `anchor_boxes` 的相同。
|
||
|
||
**异常：**
|
||
|
||
- **KeyError** - `mode` 不是'iou'或'iof'。 |