Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A 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()
H A 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
H A Dnon_max_suppression_op_test.cc34 void MakeOp(float iou_threshold) { in MakeOp()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A 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 …]
H A 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
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dnon_max_suppression_test.cc70 explicit NonMaxSuppressionV4OpModel(const float iou_threshold, in NonMaxSuppressionV4OpModel()
160 explicit NonMaxSuppressionV5OpModel(const float iou_threshold, in NonMaxSuppressionV5OpModel()
H A Dnon_max_suppression.cc222 const float iou_threshold = *GetTensorData<float>(input_iou_threshold); in Eval() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dimage_ops.cc153 ShapeHandle iou_threshold; in NMSShapeFn() local
176 ShapeHandle iou_threshold; in SoftNMSShapeFn() local
983 ShapeHandle iou_threshold; in __anon269972ec1a02() local
/aosp_15_r20/external/armnn/python/pyarmnn/examples/object_detection/
H A Dyolo.py47 def yolo_processing(output: np.ndarray, confidence_threshold=0.40, iou_threshold=0.40): argument
/aosp_15_r20/external/google-cloud-java/java-automl/proto-google-cloud-automl-v1/src/main/proto/google/cloud/automl/v1/
H A Ddetection.proto58 float iou_threshold = 1; field
/aosp_15_r20/external/googleapis/google/cloud/automl/v1/
H A Ddetection.proto58 float iou_threshold = 1; field
/aosp_15_r20/external/armnn/tests/TfLiteYoloV3Big-Armnn/
H A DNMS.hpp17 float iou_threshold{0.8f}; /**< Inclusion threshold for Intersection-Over-Union */ member
/aosp_15_r20/external/google-cloud-java/java-automl/proto-google-cloud-automl-v1beta1/src/main/proto/google/cloud/automl/v1beta1/
H A Ddetection.proto85 float iou_threshold = 1; field
/aosp_15_r20/external/googleapis/google/cloud/automl/v1beta1/
H A Ddetection.proto85 float iou_threshold = 1; field
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcustom_op_db.py415 def _(boxes, scores, iou_threshold): argument
426 def numpy_nms_vmap(info, in_dims, boxes, scores, iou_threshold): argument
/aosp_15_r20/external/google-cloud-java/java-datalabeling/proto-google-cloud-datalabeling-v1beta1/src/main/proto/google/cloud/datalabeling/v1beta1/
H A Devaluation.proto93 float iou_threshold = 1; field
/aosp_15_r20/external/googleapis/google/cloud/datalabeling/v1beta1/
H A Devaluation.proto93 float iou_threshold = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/evaluation/stages/utils/
H A Dimage_metrics.h95 float iou_threshold = 0.5; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dnms_utils.cc47 Value iou_threshold = func_.getArgument(3); in RewriteFunc() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dnon_max_suppression.h89 const float iou_threshold, in NonMaxSuppression()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/evaluation/proto/
H A Devaluation_stages.proto276 optional float iou_threshold = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A 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
H A 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

12