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

fix bug of output order

This commit is contained in:
mengyuanli 2021-07-16 10:52:15 +08:00
parent 580981a38a
commit d896a8c805
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression_fp32.h
View File

@ -77,7 +77,9 @@ class NMSBox {
}
area_ = (y2_ - y1_) * (x2_ - x1_);
}
inline bool operator<(const NMSBox &box) const { return score_ < box.score_; }
inline bool operator<(const NMSBox &box) const {
return score_ < box.score_ || (score_ == box.score_ && index_ > box.index_);
}
public:
float score_;