Skip to content

Commit d115c16

Browse files
author
Matevz Morato
committed
Add filter bindings
1 parent 5aac382 commit d115c16

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

bindings/python/src/pipeline/datatype/StereoDepthConfigBindings.cpp

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ void bind_stereodepthconfig(pybind11::module& m, void* pCallstack) {
5050
py::class_<StereoDepthConfig::CensusTransform> censusTransform(stereoDepthConfig, "CensusTransform", DOC(dai, StereoDepthConfig, CensusTransform));
5151
py::enum_<StereoDepthConfig::CensusTransform::KernelSize> censusTransformKernelSize(
5252
censusTransform, "KernelSize", DOC(dai, StereoDepthConfig, CensusTransform, KernelSize));
53-
53+
py::enum_<StereoDepthConfig::PostProcessing::Filter> filterEnum(postProcessing, "Filter", DOC(dai, StereoDepthConfig, PostProcessing, Filter));
5454
///////////////////////////////////////////////////////////////////////
5555
///////////////////////////////////////////////////////////////////////
5656
///////////////////////////////////////////////////////////////////////
@@ -82,6 +82,13 @@ void bind_stereodepthconfig(pybind11::module& m, void* pCallstack) {
8282
.value("FOOT", StereoDepthConfig::AlgorithmControl::DepthUnit::FOOT, DOC(dai, StereoDepthConfig, AlgorithmControl, DepthUnit, FOOT))
8383
.value("CUSTOM", StereoDepthConfig::AlgorithmControl::DepthUnit::CUSTOM, DOC(dai, StereoDepthConfig, AlgorithmControl, DepthUnit, CUSTOM));
8484

85+
filterEnum.value("NONE", StereoDepthConfig::PostProcessing::Filter::NONE, DOC(dai, StereoDepthConfig, PostProcessing, Filter, NONE))
86+
.value("DECIMATION", StereoDepthConfig::PostProcessing::Filter::DECIMATION, DOC(dai, StereoDepthConfig, PostProcessing, Filter, DECIMATION))
87+
.value("SPECKLE", StereoDepthConfig::PostProcessing::Filter::SPECKLE, DOC(dai, StereoDepthConfig, PostProcessing, Filter, SPECKLE))
88+
.value("MEDIAN", StereoDepthConfig::PostProcessing::Filter::MEDIAN, DOC(dai, StereoDepthConfig, PostProcessing, Filter, MEDIAN))
89+
.value("SPATIAL", StereoDepthConfig::PostProcessing::Filter::SPATIAL, DOC(dai, StereoDepthConfig, PostProcessing, Filter, SPATIAL))
90+
.value("TEMPORAL", StereoDepthConfig::PostProcessing::Filter::TEMPORAL, DOC(dai, StereoDepthConfig, PostProcessing, Filter, TEMPORAL));
91+
8592
algorithmControl.def(py::init<>())
8693
.def_readwrite("depthAlign", &StereoDepthConfig::AlgorithmControl::depthAlign, DOC(dai, StereoDepthConfig, AlgorithmControl, depthAlign))
8794
.def_readwrite("depthUnit", &StereoDepthConfig::AlgorithmControl::depthUnit, DOC(dai, StereoDepthConfig, AlgorithmControl, depthUnit))

0 commit comments

Comments
 (0)