ml-intern
File size: 799 Bytes
fae224a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from setuptools import find_packages, setup

setup(
    name="unidepth-inference",
    version="0.1.0",
    author="bdck",
    description="Clean inference wrapper for UniDepth: image → metric depth → point cloud",
    long_description=open("README.md", encoding="utf-8").read(),
    long_description_content_type="text/markdown",
    url="https://huggingface.co/bdck/unidepth-inference",
    packages=find_packages(),
    python_requires=">=3.8",
    install_requires=[
        "torch>=2.0",
        "torchvision>=0.16",
        "numpy>=1.21",
        "Pillow>=9.0",
        "huggingface_hub>=0.20",
        "safetensors>=0.4",
    ],
    extras_require={
        "full": ["unidepth @ git+https://github.com/lpiccinelli-eth/UniDepth.git"],
        "dev": ["pytest", "black", "isort"],
    },
)