!7883 fix nms calc error

Merge pull request !7883 from zhaozhenlong/lite/issue/nms_result_wrong
4 years ago · ed17a1955b
parent 427599b2df c0bbc40533
commit ed17a1955b
1 changed files with 1 additions and 1 deletions
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression.cc
@ -183,7 +183,7 @@ int NonMaxSuppressionCPUKernel::Run() {
          float intersec_x1 = std::max(cand.x1_, box.x1_);
          float intersec_x2 = std::min(cand.x2_, box.x2_);
          float intersec_y1 = std::max(cand.y1_, box.y1_);
-          float intersec_y2 = std::max(cand.y2_, box.y2_);
+          float intersec_y2 = std::min(cand.y2_, box.y2_);
          const float intersec_area =
            std::max(intersec_x2 - intersec_x1, 0.0f) * std::max(intersec_y2 - intersec_y1, 0.0f);
          if (intersec_area <= 0.0f) {