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

[Model] add PFLD model (#433)

Browse Source 
* support face alignment PFLD

* add PFLD demo

* fixed FaceAlignmentResult

* fixed bugs

* fixed img size

* fixed readme

* deal with comments

* fixed readme

* add pfld testcase

* update infer.py

* add gflags for example

* update c++ readme

* add gflags in example

* fixed for ci

* fixed gflags.cmake

* deal with comments

* update infer demo

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
WJJ1995
2022-11-02 11:50:16 +08:00
committed by GitHub
parent 7e64f4088f
commit 9437dec9f5
33 changed files with 1059 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
fastdeploy/vision/vision_pybind.cc
View File

@@ -21,6 +21,7 @@ void BindClassification(pybind11::module& m);
void BindSegmentation(pybind11::module& m);
void BindMatting(pybind11::module& m);
void BindFaceDet(pybind11::module& m);
void BindFaceAlign(pybind11::module& m);
void BindFaceId(pybind11::module& m);
void BindOcr(pybind11::module& m);
void BindTracking(pybind11::module& m);
@@ -83,6 +84,18 @@ void BindVision(pybind11::module& m) {
.def("__repr__", &vision::FaceDetectionResult::Str)
.def("__str__", &vision::FaceDetectionResult::Str);
pybind11::class_<vision::FaceAlignmentResult>(m, "FaceAlignmentResult")
.def(pybind11::init())
.def_readwrite("landmarks", &vision::FaceAlignmentResult::landmarks)
.def("__repr__", &vision::FaceAlignmentResult::Str)
.def("__str__", &vision::FaceAlignmentResult::Str);
pybind11::class_<vision::FaceRecognitionResult>(m, "FaceRecognitionResult")
.def(pybind11::init())
.def_readwrite("embedding", &vision::FaceRecognitionResult::embedding)
.def("__repr__", &vision::FaceRecognitionResult::Str)
.def("__str__", &vision::FaceRecognitionResult::Str);
pybind11::class_<vision::SegmentationResult>(m, "SegmentationResult")
.def(pybind11::init())
.def_readwrite("label_map", &vision::SegmentationResult::label_map)
@@ -91,12 +104,6 @@ void BindVision(pybind11::module& m) {
.def("__repr__", &vision::SegmentationResult::Str)
.def("__str__", &vision::SegmentationResult::Str);
pybind11::class_<vision::FaceRecognitionResult>(m, "FaceRecognitionResult")
.def(pybind11::init())
.def_readwrite("embedding", &vision::FaceRecognitionResult::embedding)
.def("__repr__", &vision::FaceRecognitionResult::Str)
.def("__str__", &vision::FaceRecognitionResult::Str);
pybind11::class_<vision::MattingResult>(m, "MattingResult")
.def(pybind11::init())
.def_readwrite("alpha", &vision::MattingResult::alpha)
@@ -122,6 +129,7 @@ void BindVision(pybind11::module& m) {
BindClassification(m);
BindSegmentation(m);
BindFaceDet(m);
BindFaceAlign(m);
BindFaceId(m);
BindMatting(m);
BindOcr(m);