prepare_foundationstereo.md

August 30, 2025 ยท View on GitHub

1. Environment Installation

1. conda env create -f [environment.yml](https://github.com/NVlabs/FoundationStereo/blob/master/environment.yml)
2. conda run -n foundation_stereo pip install flash-attn
3. conda activate foundation_stereo

2. Training on SceneFlow Dataset

export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
torchrun --nnodes=1 --nproc_per_node=8 --rdzv_backend=c10d --rdzv_endpoint=localhost:23456 tools/train.py --dist_mode --cfg_file cfgs/foundationstereo/foundationstereo_sceneflow

3. Evaluation

python tools/eval.py --cfg_file cfgs/foundationstereo/foundationstereo_sceneflow --eval_data_cfg_file cfgs/sceneflow_eval.yaml --pretrained_model your_pretrained_ckpt_path

Our Reproduced Results

ModelOriginal PaperOursConfiguration
FoundationStereo0.330.34foundationstereo_sceneflow.yaml

Access our checkpoint: BaiduDrive or Google Drive