Merge pull request #61 from MirrorYuChen/master

fix bug in NMS
Star-Clouds · Mar 5, 2020 · c932596 · c932596
2 parents 7492696 + b4d5b8a
commit c932596
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/prj-mnn/cpp/centerface.cpp b/prj-mnn/cpp/centerface.cpp
@@ -1,5 +1,6 @@
 #include "centerface.h"
 
+#include <algorithm>
 #include <iostream>
 #include <string>
 
@@ -133,6 +134,10 @@ int Centerface::Detect(const cv::Mat & img_src, std::vector<FaceInfo>* faces) {
 			faces_tmp.push_back(face_info);
 		}
 	}
+    std::sort(faces_tmp.begin(), faces_tmp.end(),
+    [](const FaceInfo& a, const FaceInfo& b) {
+        return a.score_ < b.score_;
+    });
 	NMS(faces_tmp, faces, nmsThreshold_);
 	std::cout << "end detect." << std::endl;
 	return 0;