StereoMatching_Compare_Demo / requirements.txt
shriarul5273's picture
Implemented a Gradio application for comparing stereo matching algorithms.
6dbd5d6
raw
history blame contribute delete
779 Bytes
# FoundationStereo Gradio Application Requirements
# Based on foundation_stereo conda environment
# Core PyTorch and deep learning
torch
torchvision==0.19.1
torchaudio==2.4.1
xformers==0.0.28.post1
# Computer vision and image processing
scikit-image
opencv-contrib-python
imageio
imgaug
albumentations
# Configuration and utilities
omegaconf
pyyaml
ruamel.yaml
# Scientific computing
scipy
numpy
scikit-learn
joblib
# Deep learning utilities
timm
einops
transformations
# 3D processing and visualization
open3d
trimesh
# Model and data utilities
huggingface-hub
gdown
# Development and notebooks
jupyterlab
ninja
# Web interface (Gradio)
gradio>=4.0.0
spaces # For Hugging Face Spaces ZeroGPU support
# Additional dependencies for the Gradio app
matplotlib
pillow
tqdm