Sapiens Depth

The SapiensDepth class provides a wrapper for the SapiensDepth model, which estimates depth maps from RGB images.

This model is specifically trained for images with humans as the primary subject.

boolean

default:"False"

If True, inference call is run on the local VM, else offloaded onto GRID-Cortex. Defaults to False.

This model is currently not available via Cortex.

np.ndarray

required

The input RGB image of shape

(M,N,3)

np.ndarray

The predicted depth map of shape

(M,N)

from grid.model.perception.depth.sapiens_depth import SapiensDepth
car = AirGenCar()

# We will be capturing an image from the AirGen simulator 
# and run model inference on it.

img =  car.getImage("front_center", "rgb").data

model = SapiensDepth(use_local = False)
result = model.run(rgbimage=img)
print(result.shape)

License
Source

This code is licensed under the CC-by-NC 4.0 License.

⌘I

from grid.model.perception.depth.sapiens_depth import SapiensDepth
car = AirGenCar()

# We will be capturing an image from the AirGen simulator 
# and run model inference on it.

img =  car.getImage("front_center", "rgb").data

model = SapiensDepth(use_local = False)
result = model.run(rgbimage=img)
print(result.shape)

Get Started

Open GRID

GRID Enterprise

Robot API

Simulation

AI Layer

Deployment

Data Generation Pipelines

FAQ