Searched refs:iou_threshold (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | non_max_suppression_fp32_tests.cc | 28 …float *score_data, int32_t max_output, float iou_threshold, float score_threshold, int center_box_…
64 float iou_threshold, float score_threshold, int center_box_point) { in Init() argument
77 iou_threshold_ = iou_threshold; in Init()
102 float iou_threshold = 0.5f; in TEST_F() local
107 …Init(box_tensor_shape, box_data, score_tensor_shape, score_data, max_output, iou_threshold, score_… in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | rpn_ops_declare.cc | 22 ATTR_MAP(NMSWithMask) = {{"iou_threshold", ATTR_DESC(iou_threshold, AnyTraits<float>())}};
|
| /third_party/gstreamer/gstplugins_bad/ext/onnx/ |
| D | gstonnxobjectdetector.h | 67 gfloat iou_threshold; member
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | math_ops.py | 4467 def __init__(self, iou_threshold=0.5): argument
4469 validator.check_value_type("iou_threshold", iou_threshold, [float], self.name)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 394 …false, "param_type": "required", "shape": "all"}], "attr": [{"name": "iou_threshold", "param_type"…
|