Datasets

Get Image Keys

Retrieve the available image keys from a HuggingFace dataset. Use these keys as values in camera_mappings when creating finetune jobs.

from qualia import Qualia

client = Qualia()

image_keys = client.datasets.get_image_keys("lerobot/pusht")

Response

DatasetImageKeys

{
"dataset_id": "lerobot/pusht",
"image_keys": [
  "observation.images.top",
  "observation.images.wrist"
]
}

The returned keys correspond to the image columns in the dataset that can be mapped to your model’s camera slots.