From 031427a3c8208e567f2c15390a666bae92f8febe Mon Sep 17 00:00:00 2001
From: Sebastian Hahta <joseha@utu.fi>
Date: Wed, 8 Jan 2020 13:20:04 +0200
Subject: [PATCH] depth to disparity

---
 python/ftl/misc.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/python/ftl/misc.py b/python/ftl/misc.py
index 5494382ca..c47a47cf2 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
-- 
GitLab