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

[Model] Add Paddle3D smoke model (#1766)

Browse Source 
* add smoke model

* add 3d vis

* update code

* update doc

* mv paddle3d from detection to perception

* update result for velocity

* update code for CI

* add set input data for TRT backend

* add serving support for smoke model

* update code

* update code

* update code

---------

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
yeliang2258
2023-04-14 16:30:56 +08:00
committed by GitHub
parent 81fbd54c9d
commit a509dd8ec1
53 changed files with 2610 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
fastdeploy/vision/vision_pybind.cc
View File

@@ -33,6 +33,7 @@ void BindHeadPose(pybind11::module& m);
void BindSR(pybind11::module& m);
void BindGeneration(pybind11::module& m);
void BindVisualize(pybind11::module& m);
void BindPerception(pybind11::module& m);
void BindVision(pybind11::module& m) {
pybind11::class_<vision::Mask>(m, "Mask")
@@ -108,6 +109,40 @@ void BindVision(pybind11::module& m) {
.def("__repr__", &vision::DetectionResult::Str)
.def("__str__", &vision::DetectionResult::Str);
pybind11::class_<vision::PerceptionResult>(m, "PerceptionResult")
.def(pybind11::init())
.def_readwrite("scores", &vision::PerceptionResult::scores)
.def_readwrite("label_ids", &vision::PerceptionResult::label_ids)
.def_readwrite("boxes", &vision::PerceptionResult::boxes)
.def_readwrite("center", &vision::PerceptionResult::center)
.def_readwrite("observation_angle",
&vision::PerceptionResult::observation_angle)
.def_readwrite("yaw_angle", &vision::PerceptionResult::yaw_angle)
.def_readwrite("velocity", &vision::PerceptionResult::velocity)
.def(pybind11::pickle(
[](const vision::PerceptionResult& d) {
return pybind11::make_tuple(d.scores, d.label_ids, d.boxes,
d.center, d.observation_angle,
d.yaw_angle, d.velocity);
},
[](pybind11::tuple t) {
if (t.size() != 7)
throw std::runtime_error(
"vision::PerceptionResult pickle with Invalid state!");
vision::PerceptionResult d;
d.scores = t[0].cast<std::vector<float>>();
d.label_ids = t[1].cast<std::vector<int32_t>>();
d.boxes = t[2].cast<std::vector<std::array<float, 7>>>();
d.center = t[3].cast<std::vector<std::array<float, 3>>>();
d.observation_angle = t[4].cast<std::vector<float>>();
d.yaw_angle = t[5].cast<std::vector<float>>();
d.velocity = t[6].cast<std::vector<std::array<float, 3>>>();
return d;
}))
.def("__repr__", &vision::PerceptionResult::Str)
.def("__str__", &vision::PerceptionResult::Str);
pybind11::class_<vision::OCRResult>(m, "OCRResult")
.def(pybind11::init())
.def_readwrite("boxes", &vision::OCRResult::boxes)
@@ -224,5 +259,6 @@ void BindVision(pybind11::module& m) {
BindSR(m);
BindGeneration(m);
BindVisualize(m);
BindPerception(m);
}
} // namespace fastdeploy