| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | non_max_suppression_test.cc | 70 const float iou_threshold = 0.5; in TEST() local
81 iou_threshold, score_threshold, /**sigma=**/ 0.0, selected_indices.data(), in TEST()
97 const float iou_threshold = 0.5; in TEST() local
107 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
115 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
126 const float iou_threshold = 0.5; in TEST() local
137 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
147 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
157 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
168 const float iou_threshold = 0.5; in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 788 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
797 iou_threshold=iou_threshold,
804 iou_threshold: iou_threshold_np
826 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
835 iou_threshold=iou_threshold,
842 iou_threshold: iou_threshold_np
868 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
877 iou_threshold=iou_threshold,
883 iou_threshold: iou_threshold_np,
911 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 73 const T iou_threshold) { in OverThreshold() argument
91 const float bt = bb * iou_threshold; in OverThreshold()
161 const float iou_threshold, const int bit_mask_len, in NMSKernel() argument
203 iou_threshold)) { in NMSKernel()
253 const float iou_threshold, int* d_selected_indices, int* h_nkeep, in NmsGpu() argument
301 iou_threshold, bit_mask_len, d_delete_mask)); in NmsGpu()
305 iou_threshold, bit_mask_len, d_delete_mask)); in NmsGpu()
551 const Tensor& iou_threshold) { in CheckValidInputs() argument
558 if (!TensorShapeUtils::IsScalar(iou_threshold.shape())) { in CheckValidInputs()
560 iou_threshold.shape().DebugString(), in CheckValidInputs()
[all …]
|
| D | non_max_suppression_op.cc | 307 const float score_threshold, const float iou_threshold, in DoNMSPerClass() argument
362 if (iou > iou_threshold) { in DoNMSPerClass()
444 const float score_threshold, const float iou_threshold, in BatchedNonMaxSuppressionOp() argument
478 num_classes, size_per_class, score_threshold, iou_threshold, in BatchedNonMaxSuppressionOp()
635 const Tensor& iou_threshold = context->input(3); in Compute() local
636 OP_REQUIRES(context, TensorShapeUtils::IsScalar(iou_threshold.shape()), in Compute()
638 iou_threshold.shape().DebugString())); in Compute()
639 const T iou_threshold_val = iou_threshold.scalar<T>()(); in Compute()
681 const Tensor& iou_threshold = context->input(3); in Compute() local
682 OP_REQUIRES(context, TensorShapeUtils::IsScalar(iou_threshold.shape()), in Compute()
[all …]
|
| D | non_max_suppression_op.h | 31 float iou_threshold, float score_threshold,
55 const float iou_threshold, int* d_selected_indices,
|
| D | non_max_suppression_op_benchmark_test.cc | 34 Tensor iou_threshold(float(0.3)); in CombinedNonMaxSuppression()
43 .Input(test::graph::Constant(g, iou_threshold)) in CombinedNonMaxSuppression()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | combined_nms_test.py | 38 iou_threshold = 0.5
47 iou_threshold, dtype=dtypes.float32, name='iou_threshold')
139 iou_threshold = 0.1
151 iou_threshold, dtype=dtypes.float32, name='iou_threshold')
161 iou_threshold=iou_threshold_tensor,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | NonMaxSuppressionV2.pbtxt | 16 name: "iou_threshold"
39 name: "iou_threshold"
75 name: "iou_threshold"
|
| D | NonMaxSuppressionV3.pbtxt | 16 name: "iou_threshold"
43 name: "iou_threshold"
83 name: "iou_threshold"
|
| D | NonMaxSuppressionV4.pbtxt | 16 name: "iou_threshold"
54 name: "iou_threshold"
105 name: "iou_threshold"
|
| D | CombinedNonMaxSuppression.pbtxt | 20 name: "iou_threshold"
70 name: "iou_threshold"
|
| D | NonMaxSuppression.pbtxt | 20 name: "iou_threshold"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | NonMaxSuppressionV2.pbtxt | 16 name: "iou_threshold"
39 name: "iou_threshold"
75 name: "iou_threshold"
|
| D | NonMaxSuppressionV3.pbtxt | 16 name: "iou_threshold"
43 name: "iou_threshold"
83 name: "iou_threshold"
|
| D | NonMaxSuppressionV4.pbtxt | 16 name: "iou_threshold"
54 name: "iou_threshold"
105 name: "iou_threshold"
|
| D | CombinedNonMaxSuppression.pbtxt | 20 name: "iou_threshold"
70 name: "iou_threshold"
|
| D | NonMaxSuppression.pbtxt | 20 name: "iou_threshold"
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | object_detection_average_precision_stage_test.cc | 156 detection_metrics.individual_average_precisions(0).iou_threshold(), 0.5); in TEST()
161 detection_metrics.individual_average_precisions(1).iou_threshold(), in TEST()
196 detection_metrics.individual_average_precisions(0).iou_threshold(), 0.5); in TEST()
198 detection_metrics.individual_average_precisions(9).iou_threshold(), 0.95); in TEST()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 3649 iou_threshold=0.5, argument
3690 iou_threshold = ops.convert_to_tensor(iou_threshold, name='iou_threshold')
3694 iou_threshold, score_threshold)
3702 iou_threshold=0.5, argument
3767 iou_threshold = ops.convert_to_tensor(iou_threshold, name='iou_threshold')
3777 iou_threshold,
4957 iou_threshold=0.5, argument
5016 iou_threshold = ops.convert_to_tensor(
5017 iou_threshold, dtype=dtypes.float32, name='iou_threshold')
5032 boxes, scores, max_output_size_per_class, max_total_size, iou_threshold,
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | non_max_suppression_test.cc | 70 explicit NonMaxSuppressionV4OpModel(const float iou_threshold, in NonMaxSuppressionV4OpModel() argument
82 input_iou_threshold_ = AddConstInput(TensorType_FLOAT32, {iou_threshold}); in NonMaxSuppressionV4OpModel()
160 explicit NonMaxSuppressionV5OpModel(const float iou_threshold, in NonMaxSuppressionV5OpModel() argument
173 input_iou_threshold_ = AddConstInput(TensorType_FLOAT32, {iou_threshold}); in NonMaxSuppressionV5OpModel()
|
| D | non_max_suppression.cc | 222 const float iou_threshold = *GetTensorData<float>(input_iou_threshold); in Eval() local
261 max_output_size_value, iou_threshold, score_threshold, soft_nms_sigma, in Eval()
279 max_output_size_value, iou_threshold, score_threshold, /**sigma=**/ 0.0, in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | non_max_suppression.h | 89 const float iou_threshold, in NonMaxSuppression() argument
144 if (iou >= iou_threshold) { in NonMaxSuppression()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_NonMaxSuppression.pbtxt | 31 name: "iou_threshold"
53 boxes, scores, max_output_size, iou_threshold)
|
| D | api_def_NonMaxSuppressionV2.pbtxt | 24 name: "iou_threshold"
55 boxes, scores, max_output_size, iou_threshold)
|
| D | api_def_NonMaxSuppressionV3.pbtxt | 24 name: "iou_threshold"
61 boxes, scores, max_output_size, iou_threshold, score_threshold)
|