from setuptools import setup, find_packages

setup(
    name="vr180-matting",
    version="0.1.0",
    description="VR180 Human Matting with Det-SAM2",
    packages=find_packages(),
    install_requires=[
        "torch>=2.0.0",
        "torchvision>=0.15.0",
        "opencv-python>=4.8.0",
        "numpy>=1.24.0",
        "pillow>=10.0.0",
        "pyyaml>=6.0",
        "ultralytics>=8.0.0",
        # "sam2>=1.0.0",  # Install separately: pip install git+https://github.com/facebookresearch/segment-anything-2.git
        "tqdm>=4.65.0",
        "psutil>=5.9.0",
        "ffmpeg-python>=0.2.0",
    ],
    dependency_links=[
        "git+https://github.com/facebookresearch/segment-anything-2.git#egg=sam2"
    ],
    entry_points={
        "console_scripts": [
            "vr180-matting=vr180_matting.main:main",
        ],
    },
    python_requires=">=3.8",
)