diff options
author | Tayo Oguntebi <tayo@google.com> | 2018-08-13 23:43:48 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-13 23:47:30 -0700 |
commit | cb53bfaf48588962f5799391d3a7a411dda72b49 (patch) | |
tree | 91ea97f0c40cb8b8203b2bbf64512510bd728e2a /tensorflow/core/kernels/non_max_suppression_op.cc | |
parent | e447ae4759317156d31a9421290716f0ffbffcd8 (diff) |
Modifies NonMaxSuppressionV4 output shape to be static.
PiperOrigin-RevId: 208600876
Diffstat (limited to 'tensorflow/core/kernels/non_max_suppression_op.cc')
-rw-r--r-- | tensorflow/core/kernels/non_max_suppression_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/non_max_suppression_op.cc b/tensorflow/core/kernels/non_max_suppression_op.cc index c7d0d4de0d..5d9257e20b 100644 --- a/tensorflow/core/kernels/non_max_suppression_op.cc +++ b/tensorflow/core/kernels/non_max_suppression_op.cc @@ -126,7 +126,7 @@ void DoNonMaxSuppressionOp( const Tensor& max_output_size, const float score_threshold, const std::function<bool(int, int)>& suppress_check_fn, bool pad_to_max_output_size = false, int* ptr_num_valid_outputs = nullptr) { - const int output_size = std::min(max_output_size.scalar<int>()(), num_boxes); + const int output_size = max_output_size.scalar<int>()(); std::vector<float> scores_data(num_boxes); std::copy_n(scores.flat<float>().data(), num_boxes, scores_data.begin()); |