Fixed a bug with inconsistent inference results

wang-xinyu · Mar 24, 2024 · 6d9538b · 6d9538b
1 parent f12bbe0
commit 6d9538b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/yolov8/src/postprocess.cpp b/yolov8/src/postprocess.cpp
@@ -45,6 +45,9 @@ static float iou(float lbox[4], float rbox[4]) {
 }
 
 static bool cmp(const Detection &a, const Detection &b) {
+    if (a.conf == b.conf) {
+        return a.bbox[0] > b.bbox[0]
+    }
     return a.conf > b.conf;
 }