apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-08 01:50:27 +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

3
fastdeploy/vision/facedet/contrib/yolov7face/yolov7face_pybind.cc
View File

@@ -60,7 +60,8 @@ void BindYOLOv7Face(pybind11::module& m) {
return results;
})
.def_property("conf_threshold", &vision::facedet::Yolov7FacePostprocessor::GetConfThreshold, &vision::facedet::Yolov7FacePostprocessor::SetConfThreshold)
.def_property("nms_threshold", &vision::facedet::Yolov7FacePostprocessor::GetNMSThreshold, &vision::facedet::Yolov7FacePostprocessor::SetNMSThreshold);
.def_property("nms_threshold", &vision::facedet::Yolov7FacePostprocessor::GetNMSThreshold, &vision::facedet::Yolov7FacePostprocessor::SetNMSThreshold)
.def_property("landmarks_per_face", &vision::facedet::Yolov7FacePostprocessor::GetLandmarksPerFace, &vision::facedet::Yolov7FacePostprocessor::SetLandmarksPerFace);
pybind11::class_<vision::facedet::YOLOv7Face, FastDeployModel>(m, "YOLOv7Face")
.def(pybind11::init<std::string, std::string, RuntimeOption,