ml-intern
unidepth-inference / setup.py
bdck's picture
Upload setup.py
fae224a verified
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"],
},
)