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

[Model] Update PPSeg Preprocess (#1007)

Browse Source 
* 更新PPSeg pybind and python

* 更新PPSeg pybind and python
This commit is contained in:
Zheng-Bicheng
2022-12-29 21:14:39 +08:00
committed by GitHub
parent 9795a5287d
commit dd5759bd99
8 changed files with 51 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
fastdeploy/vision/segmentation/ppseg/ppseg_pybind.cc
View File

@@ -36,9 +36,12 @@ void BindPPSeg(pybind11::module& m) {
}
return make_pair(outputs, imgs_info);;
})
.def("disable_normalize_and_permute",
&vision::segmentation::PaddleSegPreprocessor::DisableNormalizeAndPermute)
.def("disable_normalize", [](vision::segmentation::PaddleSegPreprocessor& self) {
self.DisableNormalize();
})
.def("disable_permute", [](vision::segmentation::PaddleSegPreprocessor& self) {
self.DisablePermute();
})
.def_property("is_vertical_screen",
&vision::segmentation::PaddleSegPreprocessor::GetIsVerticalScreen,
&vision::segmentation::PaddleSegPreprocessor::SetIsVerticalScreen);