File size: 904 Bytes
0b58ffc bfd7a8f 0b58ffc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
# Main dependencies torch>=2.6.0 torchvision>=0.21.0 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.8.0 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 # 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 |