bdck's picture
Upload depthpro_wrapper/__init__.py
f25b717 verified
Raw
History Blame Contribute Delete
757 Bytes
"""
DepthPro Wrapper — Image to Point Cloud
A clean, high-level Python wrapper around Apple's DepthPro model on
HuggingFace. Drop an image in → get a metric depth map and a 3D point
cloud out.
DepthPro is a zero-shot metric depth estimator (arXiv:2410.02073) that
predicts absolute-scale depth and camera focal length from a single RGB
image, without any per-scene training or known intrinsics.
"""
__version__ = "0.1.0"
from .depth_estimator import DepthProEstimator, DepthResult
from .point_cloud import depth_to_point_cloud, rgbd_to_point_cloud
from .io import load_image, save_point_cloud
__all__ = [
"DepthProEstimator",
"DepthResult",
"depth_to_point_cloud",
"rgbd_to_point_cloud",
"load_image",
"save_point_cloud",
]