StereoSA
StereoSA, a highly accurate and fast stereo matching model powered by multi-scale selective attention aggregation.
File Explorer
Download Latest Version (.zip)- __init__.py
- augmentor.py
- data_io.py
- ETH3D_loader.py
- flow_transforms.py
- KITTI2012loader.py
- KITTI2015loader.py
- kitti_dataset.py
- KITTIRawLoader.py
- listfiles.py
- middlebury_loader.py
- readpfm.py
- sceneflow_dataset.py
- vkitti_dataset.py
- kitti12_15_all.txt
- kitti12_all.txt
- kitti12_test.txt
- kitti15_all.txt
- kitti15_test.txt
- kitti15_val.txt
- kitti_raw.txt
- sceneflow_test.txt
- sceneflow_train.txt
- vkitti_train.txt
- hunter.png
- hunter_sim.png
- mygif.gif
- StereoSA.png
- stereoSA_conf.gif
- StereoSA_disp.gif
- table.png
- kitti_publisher_cuda_node.launch.py
- kitti_publisher_cuda_node.cpp
- CMakeLists.txt
- package.xml
- kitti_publisher_conf_cuda_node.launch.py
- kitti_publisher_conf_cuda_node.cpp
- CMakeLists.txt
- package.xml
- __init__.py
- loss.py
- StereoSA.py
- StereoSA_confidence.py
- StereoSA_trt.py
- submodule.py
- __init__.py
- experiment.py
- metrics.py
- visualization.py
- .gitignore
- LICENSE
- onnx_transformed.py
- onnx_transformed_conf.py
- README.md
- save_disp.py
- save_vid.py
- test_eth3d.py
- test_kitti.py
- test_mid.py
- train_kitti.py
- train_sceneflow.py
# Installation Guide
git clone https://github.com/M2219/StereoSA
Downloads the entire project code from GitHub to your computer.
cd StereoSA
Moves into the project folder you just downloaded.
2. Official Install Script
Easy Recommended- Python 3 Python is required to use pip.
pip install opencv-python
Installs the package published on PyPI directly β no need to clone the source.
pip install scikit-image
Installs the package published on PyPI directly β no need to clone the source.
pip install tensorboard
Installs the package published on PyPI directly β no need to clone the source.
pip install matplotlib
Installs the package published on PyPI directly β no need to clone the source.
Pulled directly from this repo's README.
3. CMake
Mediumcd kitti_publisher
This project's files live in a subfolder, so move into it first.
mkdir build && cd build
Creates a folder to hold the build output and moves into it.
cmake ..
Analyzes the source code and generates build configuration files (must be run inside the build folder).
make
Compiles the code based on the generated build configuration to produce an executable.