diff --git a/demo_ncnn/nanodet.cpp b/demo_ncnn/nanodet.cpp index 6b5c2cb92..c5bde52f6 100644 --- a/demo_ncnn/nanodet.cpp +++ b/demo_ncnn/nanodet.cpp @@ -160,9 +160,9 @@ void NanoDet::decode_infer(ncnn::Mat& feats, std::vector& center_pr int cur_label = 0; for (int label = 0; label < this->num_class; label++) { - if (scores[label] > score) + if (sigmoid(scores[label]) > score) { - score = scores[label]; + score = sigmoid(scores[label]); cur_label = label; } }