2.6K Star 8.6K Fork 4.9K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindspore.ops.func_iou.rst 1.48 KB
一键复制 编辑 原始数据 按行查看 历史
宦晓玲 提交于 2024-03-01 11:47 . modify the formula inconsistence 2.3

mindspore.ops.iou

.. py:function:: mindspore.ops.iou(anchor_boxes, gt_boxes, 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缩放。

    参数:
        - **anchor_boxes** (Tensor) - 预测区域,shape为 :math:`(N, 4)` 的Tensor。 :math:`N` 表示预测区域的数量, :math:`4` 表示预测区域的四个边界坐标"x0"、"y0"、"x1"和"y1"。数据类型为float16、float32或float64。
        - **gt_boxes** (Tensor) - 真实区域,shape为 :math:`(M, 4)` 的Tensor。 :math:`M` 表示地面真实区域的数量, :math:`4` 表示真实区域的四个边界坐标"x0"、"y0"、"x1"和"y1"。数据类型为float16、float32或float64。
        - **mode** (string) - 指定计算方法,现支持 ``'iou'`` (intersection over union)或 ``'iof'`` (intersection over foreground)模式。默认值: ``'iou'`` 。

    返回:
        Tensor,IOU/IOF的计算结果,shape为 :math:`(M, N)` 的Tensor,数据类型与 `anchor_boxes` 的相同。

    异常:
        - **KeyError** - `mode` 不是 ``'iou'`` 或 ``'iof'`` 。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.q1

搜索帮助

0d507c66 1850385 C8b1a773 1850385