| # Main dependencies | |
| torch>=2.6.0 | |
| torchvision>=0.21.0 | |
| lightning>=2.2.4 | |
| numpy>=1.26.4 | |
| matplotlib>=3.8.0 | |
| pyyaml>=6.0.0 | |
| # Huggingface | |
| huggingface-hub>=0.30.2 | |
| safetensors | |
| # Image processing and computer vision | |
| kornia>=0.7.2 | |
| opencv-python>=4.10.0.84 | |
| opencv-contrib-python>=4.10.0.84 | |
| einops>=0.8.0 | |
| # Custom cuda implementation /modules/cupy_module/ | |
| cupy-cuda12x>=12.0.0 # For CUDA 12.4 | |
| # Note: For cupy, you need to install the specific version for your CUDA version | |
| # Examples: | |
| # cupy-cuda11x for CUDA 11.x | |
| # cupy-cuda12x for CUDA 12.x | |
| # cupy-cuda10x for CUDA 10.x | |
| # Utilities and tools | |
| scipy>=1.7.0 | |
| tensorboard>=2.8.0 | |
| # Project-Specific Dependencies | |
| # RAFT (Flow Estimation) | |
| # Note: RAFT is included in the project code; no external installation is required. | |
| # FLOLPIPS (Quality Metrics) | |
| # Note: FLOLPIPS is included in the project code; no external installation is required. | |
| # Gradio | |
| gradio>=4.34.0 | |
| imageio>=2.34.1 | |
| imageio-ffmpeg>=0.6.0 |