From dda21b8074c91d534c8b15e094b4bc8eebb171ec Mon Sep 17 00:00:00 2001 From: FDInSky <48318485+FDInSky@users.noreply.github.com> Date: Tue, 24 Dec 2019 11:19:30 +0800 Subject: [PATCH] add an argument for iou_similarity op (#1674) * test=develop fix iou_similarity cn doc * test=develop fix something based on review --- doc/fluid/api_cn/layers_cn/iou_similarity_cn.rst | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/doc/fluid/api_cn/layers_cn/iou_similarity_cn.rst b/doc/fluid/api_cn/layers_cn/iou_similarity_cn.rst index 19342078f..4bad47777 100644 --- a/doc/fluid/api_cn/layers_cn/iou_similarity_cn.rst +++ b/doc/fluid/api_cn/layers_cn/iou_similarity_cn.rst @@ -3,7 +3,7 @@ iou_similarity ------------------------------- -.. py:function:: paddle.fluid.layers.iou_similarity(x, y, name=None) +.. py:function:: paddle.fluid.layers.iou_similarity(x, y, box_normalized=True, name=None) **IOU Similarity Operator** @@ -15,6 +15,7 @@ iou_similarity 参数: - **x** (Variable) - 框列表 :math:`X` 是二维LoDTensor,维度为 :math:`[N,4]` ,存有 :math:`N` 个框,每个框表示为 :math:`[xmin, ymin, xmax, ymax]` ,:math:`X` 的维度为 :math:`[N,4]` 。如果输入是图像特征图,:math:`[xmin, ymin]` 表示框的左上角坐标,接近坐标轴的原点。:math:`[xmax, ymax]` 表示框的右下角坐标。该张量包含批次输入的LoD信息。该批次输入的一个实例能容纳不同的项数。数据类型为float32或float64。 - **y** (Variable) - 框列表 :math:`Y` 是二维张量,存有 :math:`M` 个框,每个框表示为 :math:`[xmin, ymin, xmax, ymax]` ,:math:`Y` 的维度为 :math:`[M,4]`。如果输入是图像特征图,:math:`[xmin, ymin]` 表示框的左上角坐标,接近坐标轴的原点。:math:`[xmax, ymax]` 表示框的右下角坐标。数据类型为float32或float64。 + - **box_normalized** (bool) - 先验框坐标是否正则化,即是否在[0, 1]区间内。默认值为true 返回:维度为 :math:`[N,M]` 的LoDTensor,代表每一对iou分数,数据类型与 :math:`X` 相同 -- GitLab