diff --git a/python/ftl/misc.py b/python/ftl/misc.py index 5494382ca1376c8bf96a9d30d7d863d8bfc9eff9..c47a47cf2c3ca2680b4031a56c42eec83fd99a7d 100644 --- a/python/ftl/misc.py +++ b/python/ftl/misc.py @@ -8,6 +8,13 @@ def disparity_to_depth(disparity, camera, max_depth=10.0, invalid_value=0.0): depth[depth > max_depth] = invalid_value return depth +def depth_to_disparity(depth, camera, invalid_value=0.0): + invalid = depth == 0.0 + depth[invalid] = 1.0 + disparity = ((camera.fx * camera.baseline) / depth) + camera.doff + disparity[invalid] = invalid_value + return disparity + from enum import IntEnum # components/codecs/include/ftl/codecs/hevc.hpp