File size: 486 Bytes
87608ea
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
"""Top-level public API for PXDepth monocular depth estimation.

Importing this package exposes the :class:`PXDepth` model without pulling
evaluation entry points into user code. The model accepts RGB
tensors and returns normalized depth plus a finite-depth probability map.
"""

from .build import build_model
from .model import PXDepth
from .registry import ENCODERS, MODELS, PREDICTORS

__all__ = [
    "PXDepth",
    "build_model",
    "MODELS",
    "ENCODERS",
    "PREDICTORS",
]