From 4d9c5bbe60d680d262eb9c6cd2187c0d84c6e147 Mon Sep 17 00:00:00 2001
From: shaoshengsong <30892500+shaoshengsong@users.noreply.github.com>
Date: Sat, 8 Oct 2022 21:18:06 +0800
Subject: [PATCH] Update postprocess.cc

fix score bug. original version score like 0.99 or 1.0 after like end of article https://flyfish.blog.csdn.net/article/details/127096511
---
 rknn/rknn_api/examples/rknn_yolov5_demo/src/postprocess.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/rknn/rknn_api/examples/rknn_yolov5_demo/src/postprocess.cc b/rknn/rknn_api/examples/rknn_yolov5_demo/src/postprocess.cc
index 793efd4..3dd584c 100644
--- a/rknn/rknn_api/examples/rknn_yolov5_demo/src/postprocess.cc
+++ b/rknn/rknn_api/examples/rknn_yolov5_demo/src/postprocess.cc
@@ -220,7 +220,9 @@ static int process(uint8_t* input, int* anchor, int grid_h, int grid_w, int heig
               maxClassProbs = prob;
             }
           }
-          objProbs.push_back(sigmoid(deqnt_affine_to_f32(maxClassProbs, zp, scale)));
+          float deqnt_tmp_a = sigmoid(deqnt_affine_to_f32(box_confidence, zp, scale));
+          float deqnt_tmp_b = sigmoid(deqnt_affine_to_f32(maxClassProbs, zp, scale));
+          objProbs.push_back(deqnt_tmp_a * deqnt_tmp_b);
           classId.push_back(maxClassId);
           validCount++;
         }