更新pybind

younghuvee · Feb 15, 2023 · 8faca05 · 8faca05
1 parent 8c42b70
commit 8faca05
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/fastdeploy/vision/keypointdet/pptinypose/pptinypose_pybind.cc b/fastdeploy/vision/keypointdet/pptinypose/pptinypose_pybind.cc
@@ -36,6 +36,14 @@ void BindPPTinyPose(pybind11::module& m) {
             self.Predict(&mat, &res, detection_result);
             return res;
           })
+      .def("disable_normalize",
+           [](vision::keypointdetection::PPTinyPose& self) {
+             self.DisableNormalize();
+           })
+      .def("disable_permute",
+           [](vision::keypointdetection::PPTinyPose& self) {
+             self.DisablePermute();
+           })
       .def_readwrite("use_dark",
                      &vision::keypointdetection::PPTinyPose::use_dark);
 }