Merge pull request #14085 from jerrywgz/fix_generate_proposals_op

[1.1] fix erase end in generate proposals op
6 years ago · a3efba176c
parent b9ae1c49f8 de2f965c9b
commit a3efba176c
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/detection/generate_proposals_op.cc
+++ b/paddle/fluid/operators/detection/generate_proposals_op.cc
@ -284,7 +284,7 @@ static inline Tensor NMS(const platform::DeviceContext &ctx, Tensor *bbox,
      selected_indices.push_back(idx);
      ++selected_num;
    }
-    sorted_indices.erase(sorted_indices.end());
+    sorted_indices.erase(sorted_indices.end() - 1);
    if (flag && eta < 1 && adaptive_threshold > 0.5) {
      adaptive_threshold *= eta;
    }