Merge pull request #14085 from jerrywgz/fix_generate_proposals_op

JiabinYang · web-flow · commit a3efba176c65 · 2018-10-26T19:58:26.000+08:00
[1.1] fix erase end in generate proposals op
diff --git 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;
     }

Original file line number	Diff line number	Diff line change
`@@ -284,7 +284,7 @@ static inline Tensor NMS(const platform::DeviceContext &ctx, Tensor *bbox,`
`284`	`284`	`selected_indices.push_back(idx);`
`285`	`285`	`++selected_num;`
`286`	`286`	`}`
`287`		`- sorted_indices.erase(sorted_indices.end());`
	`287`	`+ sorted_indices.erase(sorted_indices.end() - 1);`
`288`	`288`	`if (flag && eta < 1 && adaptive_threshold > 0.5) {`
`289`	`289`	`adaptive_threshold *= eta;`
`290`	`290`	`}`