vhad/openpose
1
0
Fork 0
mirror of https://github.com/gosticks/openpose.git synced 2026-08-12 20:30:20 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added functions for pose params (#1413)

Browse Source
This commit is contained in:
Raaj
2019-11-10 11:31:59 -05:00
committed by Gines
parent 91265c88d2
commit e31bdcd866
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/openpose/openpose_python.cpp
+15
View File
@@ -293,6 +293,21 @@ namespace op
m.def("get_gpu_number", &getGpuNumber, "Get Total GPU");
m.def("get_images_on_directory", &getImagesFromDirectory, "Get Images On Directory");
// Pose Mapping
// Code example in doc/output.md, section Keypoint Ordering in C++/Python
m.def("getPoseBodyPartMapping", &getPoseBodyPartMapping, "getPoseBodyPartMapping");
m.def("getPoseNumberBodyParts", &getPoseNumberBodyParts, "getPoseNumberBodyParts");
m.def("getPosePartPairs", &getPosePartPairs, "getPosePartPairs");
m.def("getPoseMapIndex", &getPoseMapIndex, "getPoseMapIndex");
py::enum_<PoseModel>(m, "PoseModel", py::arithmetic())
.value("BODY_25", PoseModel::BODY_25)
.value("COCO_18", PoseModel::COCO_18)
.value("MPI_15", PoseModel::MPI_15)
.value("MPI_15_4", PoseModel::MPI_15_4)
.value("BODY_25B", PoseModel::BODY_25B)
.value("BODY_135", PoseModel::BODY_135)
.export_values();
// OpenposePython
py::class_<WrapperPython>(m, "WrapperPython")
.def(py::init<>())