Awesome
Selective-Stereo (CVPR 2024 Highlight)
This repository contains the source code for our paper:
Selective-Stereo: Adaptive Frequency Information Selection for Stereo Matching
CVPR 2024 Highlight
Xianqi Wang, Gangwei Xu, Hao Jia, Xin Yang
@inproceedings{wang2024selective,
title={Selective-stereo: Adaptive frequency information selection for stereo matching},
author={Wang, Xianqi and Xu, Gangwei and Jia, Hao and Yang, Xin},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={19701--19710},
year={2024}
}
Demo
Pretrained models can be downloaded from Google Drive.
To evaluate SceneFlow, run
python evaluate_stereo.py --restore_ckpt ./pretrained_models/sceneflow.pth
To predict Middlebury, run
python demo_imgs.py --restore_ckpt ./pretrained_models/middlebury_finetune.pth --valid_iters 80 --max_disp 768
Environment
conda create -n Selective_Stereo python=3.8
conda activate Selective_Stereo
pip install torch torchvision
pip install tqdm timm==0.5.4
pip install opencv-python scikit-image matplotlib
pip install opt_einsum
Required Data
- SceneFlow
- KITTI
- ETH3D
- Middlebury
- TartanAir
- CREStereo Dataset
- FallingThings
- InStereo2K
- Sintel Stereo
- HR-VS
Training
To train on SceneFlow, run
python train_stereo.py --logdir ./checkpoints/sceneflow
To train on KITTI, run
python train_stereo.py --restore_ckpt ./pretrained_models/sceneflow.pth --train_datasets kitti --num_steps 50000
To train on ETH3D, run
python train_stereo.py --restore_ckpt ./pretrained_models/sceneflow.pth --train_datasets eth3d_train --num_steps 300000 --image_size 384 512
python train_stereo.py --restore_ckpt ./pretrained_models/eth3d_train.pth --tarin_datasets eth3d_finetune --num_steps 90000
To train on Middlebury, run
python train_stereo.py --restore_ckpt ./pretrained_models/sceneflow.pth --train_datasets middlebury_train --num_steps 200000 --image_size 384 512
python train_stereo.py --restore_ckpt ./pretrained_models/middlebury_train.pth --tarin_datasets middlebury_finetune --num_steps 100000 --image_size 384 768 --max_disp 768