p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix nms calculate iou issue

This commit is contained in:
zhaozhenlong 2020-10-28 09:52:13 +08:00
parent cb5e33be88
commit c0bbc40533
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression.cc
View File

@ -183,7 +183,7 @@ int NonMaxSuppressionCPUKernel::Run() {
float intersec_x1 = std::max(cand.x1_, box.x1_); float intersec_x1 = std::max(cand.x1_, box.x1_);
float intersec_x2 = std::min(cand.x2_, box.x2_); float intersec_x2 = std::min(cand.x2_, box.x2_);
float intersec_y1 = std::max(cand.y1_, box.y1_); 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 = const float intersec_area =
std::max(intersec_x2 - intersec_x1, 0.0f) * std::max(intersec_y2 - intersec_y1, 0.0f); std::max(intersec_x2 - intersec_x1, 0.0f) * std::max(intersec_y2 - intersec_y1, 0.0f);
if (intersec_area <= 0.0f) { if (intersec_area <= 0.0f) {