Skip to content

Commit 15d2cfb

Browse files
Merge pull request #9 from Kotochleb/feature/modality-cleaning
Expose more functions of Modality and ColorHistograms
2 parents 64f1b86 + 525267a commit 15d2cfb

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/pym3t.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ PYBIND11_MODULE(_pym3t_mod, m) {
7373
.def_property_readonly("set_up", &Tracker::set_up)
7474
.def_property_readonly("optimizers", &Tracker::optimizer_ptrs)
7575
.def_property_readonly("modalities", &Tracker::modality_ptrs)
76+
.def_property_readonly("color_histograms", &Tracker::color_histograms_ptrs)
7677
;
7778

7879
//--------------------------//
@@ -368,9 +369,12 @@ PYBIND11_MODULE(_pym3t_mod, m) {
368369
// Modality -> not constructible, just to enable automatic downcasting and binding of child classes
369370
py::class_<Modality, PyModality, std::shared_ptr<Modality>>(m, "Modality")
370371
.def("SetUp", &Modality::SetUp)
372+
.def("StartModality", &Modality::StartModality)
371373
.def_property_readonly("gradient", &Modality::gradient)
372374
.def_property_readonly("hessian", &Modality::hessian)
373375
.def_property("name", &Modality::name, &Modality::set_name)
376+
.def_property_readonly("body", &Modality::body_ptr)
377+
.def_property_readonly("color_histograms", &Modality::color_histograms_ptr)
374378
;
375379

376380
// RegionModality

0 commit comments

Comments
 (0)