| # 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 |
