apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Bug fix]add yolov7face landmarks (#1297)

Browse Source 
* add yolov7face benchmark

* fix review problem

* fix review problems
This commit is contained in:
CoolCola
2023-02-14 18:36:28 +08:00
committed by GitHub
parent c25d1cc1bc
commit a5d23c57d0
4 changed files with 48 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
fastdeploy/vision/facedet/contrib/yolov7face/postprocessor.h
View File

@@ -56,11 +56,19 @@ class FASTDEPLOY_DECL Yolov7FacePostprocessor{
/// Get nms_threshold, default 0.45
float GetNMSThreshold() const { return nms_threshold_; }
/// Set landmarks_per_face, default 5
void SetLandmarksPerFace(const int& landmarks_per_face) {
landmarks_per_face_ = landmarks_per_face;
}
/// Get landmarks_per_face, default 5
int GetLandmarksPerFace() const { return landmarks_per_face_; }
protected:
float conf_threshold_;
float nms_threshold_;
bool multi_label_;
float max_wh_;
int landmarks_per_face_;
};
} // namespace facedet