| # =============================== | |
| # BackgroundFX Pro — Requirements | |
| # Python 3.10 + CUDA 12.1 + Torch 2.5.1 | |
| # SAM2 + MatAnyone Integration with Audio Support | |
| # =============================== | |
| # --- PyTorch Index (CUDA 12.1) --- | |
| --extra-index-url https://download.pytorch.org/whl/cu121 | |
| # --- PyTorch (Must be first) --- | |
| torch==2.5.1 | |
| torchvision==0.20.1 | |
| # --- Core deps (before models) --- | |
| numpy>=1.24.4 | |
| einops==0.6.0 | |
| Pillow==10.0.0 | |
| protobuf==3.20.3 | |
| # --- Image / Video Processing --- | |
| opencv-python-headless==4.7.0.72 | |
| imageio==2.25.0 | |
| imageio-ffmpeg==0.4.7 | |
| moviepy==1.0.3 | |
| decord==0.6.0 | |
| scikit-image==0.19.3 | |
| # --- SAM2 Dependencies --- | |
| hydra-core==1.3.2 | |
| omegaconf==2.3.0 | |
| timm==0.9.0 | |
| pyyaml==6.0.0 | |
| matplotlib==3.5.0 | |
| iopath==0.1.10 | |
| # --- SAM2 Installation --- | |
| git+https://github.com/facebookresearch/segment-anything-2.git | |
| # --- MatAnyone Dependencies (BEFORE MatAnyone) --- | |
| huggingface-hub>=0.20.0 | |
| kornia==0.7.2 | |
| tqdm==4.66.1 | |
| scipy==1.10.1 | |
| transformers==4.30.0 | |
| accelerate==0.20.0 | |
| gradio>=3.34 | |
| xformers>=0.0.28 | |
| # --- MatAnyone Installation --- | |
| git+https://github.com/pq-yang/MatAnyone.git | |
| # --- Streamlit UI --- | |
| streamlit==1.32.0 | |
| # --- Web stack --- | |
| fastapi==0.104.0 | |
| uvicorn==0.24.0 | |
| # --- Pydantic / Typing --- | |
| pydantic==2.6.0 | |
| pydantic-core==2.16.1 | |
| annotated-types==0.6.0 | |
| typing-extensions==4.8.0 | |
| # --- Utilities --- | |
| ffmpeg-python==0.2.0 | |
| psutil==5.9.4 | |
| requests==2.31.0 | |
| scikit-learn==1.3.0 | |
| loguru==0.6.0 | |
| python-multipart==0.0.5 | |
| ninja==1.10.0 | |
| aiofiles==23.1.0 |