mli0603 / stereo-transformer

Revisiting Stereo Depth Estimation From a Sequence-to-Sequence Perspective with Transformers. (ICCV 2021 Oral)
Apache License 2.0
659 stars 107 forks source link

Depth map #68

Open chaimaOues opened 2 years ago

chaimaOues commented 2 years ago

Hello, with this code we can determinate only the image disparity. How can I convert the image disparity into depth map ? thank you

mli0603 commented 2 years ago

depth = f * b / disp, where f is focal length and b is baseline

chaimaOues commented 2 years ago

I found that for KITTI 2015 the focal length and baseline are 721 pixels and 0.54 m and i applied this formula: Depth= (54 * 721 )/ (disp_pred) I got this result: resultDepth