modify nms

Star-Clouds · Mar 5, 2020 · 980e127 · 980e127
1 parent 7492696
commit 980e127
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/prj-mnn/cpp/centerface.cpp b/prj-mnn/cpp/centerface.cpp
@@ -133,6 +133,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;