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

[Model] add pptracking model (#357)

Browse Source 
* add override mark

* delete some

* recovery

* recovery

* add tracking

* add tracking py_bind and example

* add pptracking

* add pptracking

* iomanip head file

* add opencv_video lib

* add python libs package

Signed-off-by: ChaoII <849453582@qq.com>

* complete comments

Signed-off-by: ChaoII <849453582@qq.com>

* add jdeTracker_ member variable

Signed-off-by: ChaoII <849453582@qq.com>

* add 'FASTDEPLOY_DECL' macro

Signed-off-by: ChaoII <849453582@qq.com>

* remove kwargs params

Signed-off-by: ChaoII <849453582@qq.com>

* [Doc]update pptracking docs

* delete 'ENABLE_PADDLE_FRONTEND' switch

* add pptracking unit test

* update pptracking unit test

Signed-off-by: ChaoII <849453582@qq.com>

* modify test video file path and remove trt test

* update unit test model url

* remove 'FASTDEPLOY_DECL' macro

Signed-off-by: ChaoII <849453582@qq.com>

* fix build python packages about pptracking on win32

Signed-off-by: ChaoII <849453582@qq.com>

Signed-off-by: ChaoII <849453582@qq.com>
Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
ChaoII
2022-10-26 14:27:55 +08:00
committed by GitHub
parent da7247aa41
commit ba501fd963
38 changed files with 2959 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 BindMatting(pybind11::module& m);
void BindFaceDet(pybind11::module& m);
void BindFaceId(pybind11::module& m);
void BindOcr(pybind11::module& m);
void BindTracking(pybind11::module& m);
void BindKeyPointDetection(pybind11::module& m);
#ifdef ENABLE_VISION_VISUALIZE
void BindVisualize(pybind11::module& m);
@@ -63,6 +64,15 @@ void BindVision(pybind11::module& m) {
.def("__repr__", &vision::OCRResult::Str)
.def("__str__", &vision::OCRResult::Str);
pybind11::class_<vision::MOTResult>(m, "MOTResult")
.def(pybind11::init())
.def_readwrite("boxes", &vision::MOTResult::boxes)
.def_readwrite("ids", &vision::MOTResult::ids)
.def_readwrite("scores", &vision::MOTResult::scores)
.def_readwrite("class_ids", &vision::MOTResult::class_ids)
.def("__repr__", &vision::MOTResult::Str)
.def("__str__", &vision::MOTResult::Str);
pybind11::class_<vision::FaceDetectionResult>(m, "FaceDetectionResult")
.def(pybind11::init())
.def_readwrite("boxes", &vision::FaceDetectionResult::boxes)
@@ -112,6 +122,7 @@ void BindVision(pybind11::module& m) {
BindFaceId(m);
BindMatting(m);
BindOcr(m);
BindTracking(m);
BindKeyPointDetection(m);
#ifdef ENABLE_VISION_VISUALIZE
BindVisualize(m);