> ## 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.

# OneFormer 

<Note>
  This page documents the `grid.model.*` local inference wrapper. For cloud-hosted inference via **GRID Cortex**, see the [Cortex OneFormer page](/models/cortex/oneformer).
</Note>

The `OneFormer` class provides core functionality for this module.

<ResponseField name="class OneFormer()">
  <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="mode" type="str" default="semantic">
    The mode of segmentation. Can be either "semantic" or "panoptic". Defaults to "semantic".
  </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.oneformer import OneFormer
  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 = OneFormer(use_local = False)
  result = model.run(rgbimage=img, mode="semantic")
  print(result.shape)
  ```
</RequestExample>

<Tabs>
  <Tab title="License">
    This code is licensed under the MIT License.
  </Tab>

  <Tab title="Source">
    [https://github.com/SHI-Labs/OneFormer](https://github.com/SHI-Labs/OneFormer)
  </Tab>
</Tabs>