test=develop

de2f965c · jerrywgz · 18be7256 · de2f965c
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/detection/generate_proposals_op.cc paddle/fluid/operators/detection/generate_proposals_op.cc +1 -1

未找到文件。
--- 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;
    }