From d896a8c8053e2a4e1d643b4ed796a0f256486f8a Mon Sep 17 00:00:00 2001 From: mengyuanli Date: Fri, 16 Jul 2021 10:52:15 +0800 Subject: [PATCH] fix bug of output order --- .../src/runtime/kernel/arm/fp32/non_max_suppression_fp32.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression_fp32.h b/mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression_fp32.h index bbfa7f078e2..fe7f74f626d 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression_fp32.h +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/non_max_suppression_fp32.h @@ -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_;