requirements_python.txt 672 B

123456789101112131415161718192021222324252627282930
  1. # Python 图像背景移除库依赖
  2. # FastAPI 服务器依赖
  3. fastapi>=0.104.0
  4. uvicorn[standard]>=0.24.0
  5. requests>=2.31.0
  6. websockets>=12.0
  7. aiohttp>=3.9.0
  8. # 方案1: rembg (最简单,推荐,CPU友好)
  9. rembg>=2.0.0
  10. Pillow>=9.0.0
  11. # 方案2: BiRefNet (效果最好,需要GPU)
  12. # 注意: BiRefNet 的完整依赖在 BiRefNet/requirements.txt 中
  13. # 如果需要使用 BiRefNet,请运行: pip install -r BiRefNet/requirements.txt
  14. # 主要依赖包括:
  15. # - torch>=2.5.0
  16. # - torchvision
  17. # - numpy<2
  18. # - opencv-python
  19. # - timm
  20. # - scipy
  21. # - scikit-image
  22. # - kornia
  23. # - einops
  24. # - transformers (用于从 HuggingFace 加载模型)
  25. # - huggingface-hub>0.25
  26. # - accelerate