Browse Source

!11576 update the example of BoundingBoxDecode.

From: @wangshuide2020
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui
tags/v1.1.1
mindspore-ci-bot Gitee 5 years ago
parent
commit
53384bc05d
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/ops/operations/other_ops.py

+ 4
- 4
mindspore/ops/operations/other_ops.py View File

@@ -128,13 +128,13 @@ class BoundingBoxEncode(PrimitiveWithInfer):
``Ascend`` ``GPU``

Examples:
>>> anchor_box = Tensor([[4, 1, 2, 1], [2, 2, 2, 3]], mindspore.float32)
>>> groundtruth_box = Tensor([[3, 1, 2, 2], [1, 2, 1, 4]], mindspore.float32)
>>> anchor_box = Tensor([[2, 2, 2, 3], [2, 2, 2, 3]], mindspore.float32)
>>> groundtruth_box = Tensor([[1, 2, 1, 4], [1, 2, 1, 4]], mindspore.float32)
>>> boundingbox_encode = ops.BoundingBoxEncode(means=(0.0, 0.0, 0.0, 0.0), stds=(1.0, 1.0, 1.0, 1.0))
>>> output = boundingbox_encode(anchor_box, groundtruth_box)
>>> print(output)
[[ 5.0000000e-01 5.0000000e-01 -6.5504000e+04 6.9335938e-01]
[-1.0000000e+00 2.5000000e-01 0.0000000e+00 4.0551758e-01]]
[[ -1. 0.25 0. 0.40551758]
[ -1. 0.25 0. 0.40551758]]
"""

@prim_attr_register


Loading…
Cancel
Save