Fast Disparity Map Generation using Vision Mamba

This repository contains code heavily adapted from UniMatch and Vim.

Setup

To set up the environment, please follow the steps provided in the following repositories:

• UniMatch Environment Setup
• Vim Environment Setup

Note: Cuda version 11.8 is a requirement to run the mamba models.

Running Inference

To run inference for frames per second (fps), use the following command:

bash evaluate_fps.sh

To run inference for error, use the following command:

bash evaluate_error.sh

Checkpoints

final_checkpoint.pth is the model weights for the proposed model.