> ## Documentation Index
> Fetch the complete documentation index at: https://docs.generalrobotics.dev/llms.txt
> Use this file to discover all available pages before exploring further.

# Sapiens Segmentation 

The `SapiensSegmentation` class provides a wrapper for the Sapiens body-part segmentation model.

<ResponseField name="class SapiensSegmentation()">
  <ResponseField name="use_local" type="boolean" default="False">
    If True, inference call is run on the local VM, else offloaded onto GRID-Cortex. Defaults to False.
  </ResponseField>
</ResponseField>

<ResponseField name="def run()">
  <ResponseField name="rgbimage" type="np.ndarray" required="True">
    The input RGB image of shape $(M, N, 3)$.
  </ResponseField>

  <ResponseField name="Returns" type="np.ndarray">
    The predicted segmentation mask of shape $(M, N)$.
  </ResponseField>
</ResponseField>

<RequestExample>
  ```python Inference Call theme={null}
  from grid.model.perception.segmentation.sapiens_segmentation import SapiensSegmentation
  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 = SapiensSegmentation(use_local = False)
  result = model.run(rgbimage=img)
  print(result.shape)
  ```
</RequestExample>

<Tabs>
  <Tab title="License">
    This code is licensed under the CC-by-NC 4.0 License.
  </Tab>

  <Tab title="Source">
    [https://huggingface.co/facebook/sapiens-seg-0.3b](https://huggingface.co/facebook/sapiens-seg-0.3b)
  </Tab>
</Tabs>