From 742cc8c571fcfec937c8f49f87e8faa899abfd4f Mon Sep 17 00:00:00 2001
From: zhiliu6 <zhiliu6@gmail.com>
Date: Sat, 23 Nov 2019 15:55:54 +0800
Subject: [PATCH] optimize yolov3 output extraction speed. (#1398)

---
 src/layer/yolov3detectionoutput.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/layer/yolov3detectionoutput.cpp b/src/layer/yolov3detectionoutput.cpp
index 1d15375cf..11c7b043e 100644
--- a/src/layer/yolov3detectionoutput.cpp
+++ b/src/layer/yolov3detectionoutput.cpp
@@ -221,16 +221,17 @@ int Yolov3DetectionOutput::forward(const std::vector<Mat>& bottom_blobs, std::ve
 
                     // find class index with max class score
                     int class_index = 0;
-                    float class_score = 0.f;
+                    float class_score = -std::numeric_limits<float>::max();
                     for (int q = 0; q < num_class; q++)
                     {
-                        float score = sigmoid(scores.channel(q).row(i)[j]);
+                        float score = scores.channel(q).row(i)[j];
                         if (score > class_score)
                         {
                             class_index = q;
                             class_score = score;
                         }
                     }
+                    class_score = sigmoid(class_score);
 
                     //printf( "%d %f %f\n", class_index, box_score, class_score);