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