Python non_max_suppression示例

示例#1

文件： test_reduction.py 项目： ravirajpinnamaraju/ngraph

def test_non_max_suppression():

    boxes_shape = [1, 1000, 4]
    scores_shape = [1, 1, 1000]
    expected_shape = [0, 3]
    boxes_parameter = ng.parameter(boxes_shape, name='Boxes', dtype=np.float32)
    scores_parameter = ng.parameter(scores_shape, name='Scores', dtype=np.float32)

    node = ng.non_max_suppression(boxes_parameter, scores_parameter)

    assert node.get_type_name() == 'NonMaxSuppression'
    assert node.get_output_size() == 1
    assert list(node.get_output_shape(0)) == expected_shape

示例#2

文件： test_reduction.py 项目： wjhsmn424896/openvino

def test_non_max_suppression():

    boxes_shape = [1, 1000, 4]
    scores_shape = [1, 1, 1000]
    boxes_parameter = ng.parameter(boxes_shape, name="Boxes", dtype=np.float32)
    scores_parameter = ng.parameter(scores_shape, name="Scores", dtype=np.float32)

    node = ng.non_max_suppression(boxes_parameter, scores_parameter)

    assert node.get_type_name() == "NonMaxSuppression"
    assert node.get_output_size() == 3
    assert node.get_output_partial_shape(0).same_scheme(PartialShape([-1, 3]))
    assert node.get_output_partial_shape(1).same_scheme(PartialShape([-1, 3]))
    assert list(node.get_output_shape(2)) == [1]

示例#3

def test_non_max_suppression():

    boxes_shape = [1, 1000, 4]
    scores_shape = [1, 1, 1000]
    boxes_parameter = ng.parameter(boxes_shape, name="Boxes", dtype=np.float32)
    scores_parameter = ng.parameter(scores_shape, name="Scores", dtype=np.float32)

    node = ng.non_max_suppression(boxes_parameter, scores_parameter, make_constant_node(1000, np.int64))

    assert node.get_type_name() == "NonMaxSuppression"
    assert node.get_output_size() == 3
    assert node.get_output_partial_shape(0) == PartialShape([Dimension(0, 1000), Dimension(3)])
    assert node.get_output_partial_shape(1) == PartialShape([Dimension(0, 1000), Dimension(3)])
    assert list(node.get_output_shape(2)) == [1]