requirements.txt

# Gradio 和 Spaces
gradio>=4.44.0
spaces>=0.28.3

# PyTorch (使用 CUDA 12.1 版本)
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.0
torchvision==0.16.0

# 基础科学计算
numpy==1.26.4
opencv-python==4.8.1.78
pillow>=10.1.0
scipy>=1.11.4
scikit-learn>=1.3.2

# 配置管理
hydra-core==1.3.2
omegaconf==2.3.0

# 视频处理
mediapy>=1.1.9

# 3D 处理
open3d>=0.18.0

# 机器学习
transformers==4.42.4
timm>=0.9.12

# 其他工具
joblib
pandas
PyOpenGL==3.1.4
Rtree
protobuf==3.20.0
gdown

# MMCV (将在 setup.sh 中安装 full 版本)
mmcv==1.3.9