Home
last modified time | relevance | path

Searched defs:iou_threshold (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dnon_max_suppression_test.cc70 const float iou_threshold = 0.5; in TEST() local
97 const float iou_threshold = 0.5; in TEST() local
126 const float iou_threshold = 0.5; in TEST() local
168 const float iou_threshold = 0.5; in TEST() local
205 const float iou_threshold = 0.5; in TEST() local
272 const float iou_threshold = 1.0; in TEST() local
309 const float iou_threshold = 0.5; in TEST() local
/external/ComputeLibrary/src/core/CPP/kernels/
DCPPNonMaximumSuppressionKernel.cpp39 const float score_threshold, const float iou_threshold) in validate_arguments()
63 … unsigned int max_output_size, const float score_threshold, const float iou_threshold) in configure()
86 … unsigned int max_output_size, const float score_threshold, const float iou_threshold) in validate()
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc77 const T iou_threshold) { in OverThreshold()
167 const float iou_threshold, const int bit_mask_len, in NMSKernel()
480 const Tensor& iou_threshold) { in CheckValidInputs()
537 const Tensor& iou_threshold = context->input(3); in Compute() local
576 const Tensor& iou_threshold = context->input(3); in Compute() local
621 const Tensor& iou_threshold = context->input(3); in Compute() local
668 const float iou_threshold, int* d_selected_indices, int* h_nkeep, in NmsGpu()
Dnon_max_suppression_op.cc342 const float score_threshold, const float iou_threshold, in DoNMSPerClass()
467 const float score_threshold, const float iou_threshold, in BatchedNonMaxSuppressionOp()
679 const Tensor& iou_threshold = context->input(3); in Compute() local
725 const Tensor& iou_threshold = context->input(3); in Compute() local
781 const Tensor& iou_threshold = context->input(3); in Compute() local
850 const Tensor& iou_threshold = context->input(3); in Compute() local
1000 const Tensor& iou_threshold = context->input(4); in Compute() local
Dnon_max_suppression_op_test.cc34 void MakeOp(float iou_threshold) { in MakeOp()
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression_test.cc70 explicit NonMaxSuppressionV4OpModel(const float iou_threshold, in NonMaxSuppressionV4OpModel()
160 explicit NonMaxSuppressionV5OpModel(const float iou_threshold, in NonMaxSuppressionV5OpModel()
Dnon_max_suppression.cc222 const float iou_threshold = *GetTensorData<float>(input_iou_threshold); in Eval() local
/external/armnn/python/pyarmnn/examples/object_detection/
Dyolo.py47 def yolo_processing(output: np.ndarray, confidence_threshold=0.40, iou_threshold=0.40): argument
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py3756 iou_threshold=0.5, argument
3809 iou_threshold=0.5, argument
5079 iou_threshold=0.5, argument
5204 def _self_suppression(iou, _, iou_sum, iou_threshold): argument
5242 def _cross_suppression(boxes, box_slice, iou_threshold, inner_idx, tile_size): argument
5270 def _suppression_loop_body(boxes, iou_threshold, output_size, idx, tile_size): argument
5292 def cross_suppression_func(boxes, box_slice, iou_threshold, inner_idx): argument
5343 iou_threshold=0.5, argument
5438 iou_threshold=0.5, argument
5625 def suppression_loop_body(boxes, iou_threshold, output_size, idx): argument
[all …]
Dimage_ops_test.py5398 def func(boxes, scores, max_output_size, iou_threshold): argument
5477 def func(boxes, scores, max_output_size, iou_threshold, score_threshold): argument
/external/armnn/tests/TfLiteYoloV3Big-Armnn/
DNMS.hpp17 float iou_threshold{0.8f}; /**< Inclusion threshold for Intersection-Over-Union */ member
/external/tensorflow/tensorflow/core/ops/
Dimage_ops.cc153 ShapeHandle iou_threshold; in NMSShapeFn() local
176 ShapeHandle iou_threshold; in SoftNMSShapeFn() local
983 ShapeHandle iou_threshold; in __anonf6f833bb1a02() local
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/utils/
Dimage_metrics.h95 float iou_threshold = 0.5; member
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dnon_max_suppression.h89 const float iou_threshold, in NonMaxSuppression()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dnms_utils.cc47 Value iou_threshold = func_.getArgument(3); in RewriteFunc() local
/external/tensorflow/tensorflow/lite/tools/evaluation/proto/
Devaluation_stages.proto276 optional float iou_threshold = 1; field
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc4372 auto iou_threshold = in TEST_P() local
4395 const float iou_threshold; in TEST_P() member
4572 auto iou_threshold = in TEST_P() local
4595 const float iou_threshold; in TEST_P() member
Dconvert_nodes.cc5241 TRT_ShapedWeights iou_threshold = inputs.at(4).weights(); in ConvertCombinedNMS() local
5400 TRT_ShapedWeights iou_threshold = inputs.at(4).weights(); in ConvertCombinedNMS() local
/external/ComputeLibrary/arm_compute/core/
DTypes.h1145 float iou_threshold() const in iou_threshold() function