bdck commited on
Commit
f25b717
·
verified ·
1 Parent(s): 807d434

Upload depthpro_wrapper/__init__.py

Browse files
Files changed (1) hide show
  1. depthpro_wrapper/__init__.py +26 -0
depthpro_wrapper/__init__.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ DepthPro Wrapper — Image to Point Cloud
3
+
4
+ A clean, high-level Python wrapper around Apple's DepthPro model on
5
+ HuggingFace. Drop an image in → get a metric depth map and a 3D point
6
+ cloud out.
7
+
8
+ DepthPro is a zero-shot metric depth estimator (arXiv:2410.02073) that
9
+ predicts absolute-scale depth and camera focal length from a single RGB
10
+ image, without any per-scene training or known intrinsics.
11
+ """
12
+
13
+ __version__ = "0.1.0"
14
+
15
+ from .depth_estimator import DepthProEstimator, DepthResult
16
+ from .point_cloud import depth_to_point_cloud, rgbd_to_point_cloud
17
+ from .io import load_image, save_point_cloud
18
+
19
+ __all__ = [
20
+ "DepthProEstimator",
21
+ "DepthResult",
22
+ "depth_to_point_cloud",
23
+ "rgbd_to_point_cloud",
24
+ "load_image",
25
+ "save_point_cloud",
26
+ ]