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

Add YOLOv5Face model support (#38)

Browse Source 
* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

* add yolox c++/pybind and model_zoo demo

* move some helpers to private

* fixed CMakeLists.txt typos

* add normalize with alpha and beta

* add version notes for yolov5/yolov6/yolox

* add copyright to yolov5.cc

* revert normalize

* fixed some bugs in yolox

* Add YOLOv5Face Model support

* fixed examples/vision typos

* fixed runtime_option print func bugs
This commit is contained in:
DefTruth
2022-07-25 21:55:56 +08:00
committed by GitHub
parent 36fc77e6b8
commit fc71d79e58
27 changed files with 1240 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
fastdeploy/vision/vision_pybind.cc
View File

@@ -23,6 +23,7 @@ void BindPPSeg(pybind11::module& m);
void BindUltralytics(pybind11::module& m);
void BindMeituan(pybind11::module& m);
void BindMegvii(pybind11::module& m);
void BindDeepCam(pybind11::module& m);
void BindRangiLyu(pybind11::module& m);
#ifdef ENABLE_VISION_VISUALIZE
void BindVisualize(pybind11::module& m);
@@ -44,6 +45,15 @@ void BindVision(pybind11::module& m) {
.def("__repr__", &vision::DetectionResult::Str)
.def("__str__", &vision::DetectionResult::Str);
pybind11::class_<vision::FaceDetectionResult>(m, "FaceDetectionResult")
.def(pybind11::init())
.def_readwrite("boxes", &vision::FaceDetectionResult::boxes)
.def_readwrite("scores", &vision::FaceDetectionResult::scores)
.def_readwrite("landmarks", &vision::FaceDetectionResult::landmarks)
.def_readwrite("landmarks_per_face",
&vision::FaceDetectionResult::landmarks_per_face)
.def("__repr__", &vision::FaceDetectionResult::Str)
.def("__str__", &vision::FaceDetectionResult::Str);
pybind11::class_<vision::SegmentationResult>(m, "SegmentationResult")
.def(pybind11::init())
.def_readwrite("masks", &vision::SegmentationResult::masks)
@@ -57,6 +67,7 @@ void BindVision(pybind11::module& m) {
BindWongkinyiu(m);
BindMeituan(m);
BindMegvii(m);
BindDeepCam(m);
BindRangiLyu(m);
#ifdef ENABLE_VISION_VISUALIZE
BindVisualize(m);