From 525267a10c94a7412797ce1f3e82b26f57e7c89b Mon Sep 17 00:00:00 2001 From: Krzysztof Wojciechowski Date: Tue, 10 Sep 2024 13:56:49 +0000 Subject: [PATCH] Expose more functions of Modality and ColorHistograms --- src/pym3t.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/pym3t.cpp b/src/pym3t.cpp index 256a12b..ef81587 100644 --- a/src/pym3t.cpp +++ b/src/pym3t.cpp @@ -73,6 +73,7 @@ PYBIND11_MODULE(_pym3t_mod, m) { .def_property_readonly("set_up", &Tracker::set_up) .def_property_readonly("optimizers", &Tracker::optimizer_ptrs) .def_property_readonly("modalities", &Tracker::modality_ptrs) + .def_property_readonly("color_histograms", &Tracker::color_histograms_ptrs) ; //--------------------------// @@ -368,9 +369,12 @@ PYBIND11_MODULE(_pym3t_mod, m) { // Modality -> not constructible, just to enable automatic downcasting and binding of child classes py::class_>(m, "Modality") .def("SetUp", &Modality::SetUp) + .def("StartModality", &Modality::StartModality) .def_property_readonly("gradient", &Modality::gradient) .def_property_readonly("hessian", &Modality::hessian) .def_property("name", &Modality::name, &Modality::set_name) + .def_property_readonly("body", &Modality::body_ptr) + .def_property_readonly("color_histograms", &Modality::color_histograms_ptr) ; // RegionModality