Skip to main content

Lightwheel Kitchen Manipulation Task Dataset

The Lightwheel Kitchen Manipulation Task Dataset is a large-scale, open repository of robot demonstration data for imitation learning and manipulation benchmarking, collected in diverse simulated kitchen environments. The full dataset is available at here.

Overview

  • Task Coverage: 219 unique kitchen manipulation tasks in total, comprising 89 tasks from the first set and 130 from the second set, covering a wide range of skills such as appliance operation, sorting, cleaning, food preparation, and storage.
  • Multi-robot: Data is collected using four robot platforms: LeRobot ARX-X7s (X7s), Unitree G1_WBC (G1-WBC), Agilex-Piper (Piper), and Unitree G1-Controller (G1-Controller).
  • Scale: 50 demonstration episodes are provided for each (robot, task) combination, resulting in a total of 21,500 episodes (20,537,015 frames) across all tasks and robots.
  • Diversity: Episodes are gathered in realistic and varied simulated kitchens with interactive fixtures and a wide array of objects and arrangements, supporting research in generalization and robust perception and control.
  • Standardization: All data adhere to a consistent file structure and rich metadata schema, facilitating easy task filtering, loading, and benchmarking.

Please refer to the following tables for a detailed mapping of all 219 tasks, robot coverage, and precise dataset composition.

Lightwheel-Robocasa-Tasks Coverage

No.Task NameX7sG1-WBC
1AdjustToasterOvenTemperature
2AfterwashSorting
3ArrangeTea
4ArrangeVegetables
5ArrangeVegetablesSimple
6BeverageOrganization
7BowlAndCup
8BreadAndCheese
9BreadSelection
10BreadSetupSlicing
11CheesyBread
12ClearClutter
13ClearingCleaningReceptacles
14ClearingTheCuttingBoard
15CloseCabinet
16CloseDishwasher
17CloseDrawer
18CloseElectricKettleLid
19CloseFridge
20CloseMicrowave
21CloseOven
22CloseStandMixerHead
23CloseToasterOvenDoor
24CoffeeServeMug
25CoffeeSetupMug
26ColorfulSalsa
27CupcakeCleanup
28DefrostByCategory
29DessertAssembly
30DessertUpgrade
31DrawerUtensilSort
32DryDishes
33DryDrinkware
34FoodCleanup
35KettleBoiling
36MealPrepStaging
37NavigateKitchen
38OpenDishwasher
39OpenDrawer
40OpenElectricKettleLid
41OpenFridge
42OpenMicrowave
43OpenOven
44OpenStandMixerHead
45OpenToasterOvenDoor
46OrganizeBakingIngredients
47OrganizeVegetables
48PanTransfer
49PastryDisplay
50PnPCounterToCabinet
51PnPCounterToMicrowave
52PnPCounterToOven
53PnPCounterToSink
54PnPCounterToStandMixer
55PnPCounterToStove
56PnPCounterToToasterOven
57PnPOvenToCounter
58PnPSinkToCounter
59PnPStoveToCounter
60PnPToasterOvenToCounter
61PnPToasterToCounter
62PreSoakPan
63PreheatOven
64PrepMarinatingMeat
65PrepareCoffee
66PushUtensilsToSink
67QuickThaw
68RestockPantry
69ShakerShuffle
70SizeSorting
71SlideOvenRack
72SlideToasterOvenRack
73SnackSorting
74StackBowlsInSink
75StartCoffeeMachine
76StockingBreakfastFoods
77ThawInSink
78TurnOffMicrowave
79TurnOffSinkFaucet
80TurnOffStove
81TurnOnElectricKettle
82TurnOnMicrowave
83TurnOnSinkFaucet
84TurnOnStove
85TurnOnToaster
86TurnOnToasterOven
87TurnSinkSpout
88VeggieDipPrep
89WarmCroissant

Lightwheel-LIBERO-Tasks Coverage

No.Task NamePiperX7SG1-Controller
1L10K3TurnOnTheStoveAndPutTheMokaPotOnIt
2L10K4PutTheBlackBowlInTheBottomDrawerOfTheCabinetAndCloseIt
3L10K6PutTheYellowAndWhiteMugInTheMicrowaveAndCloseIt
4L10K8PutBothMokaPotsOnTheStove
5L10L1PutBothTheAlphabetSoupAndTheCreamCheeseBoxInTheBasket
6L10L2PutBothTheAlphabetSoupAndTheTomatoSauceInTheBasket
7L10L2PutBothTheCreamCheeseBoxAndTheButterInTheBasket
8L10L5PutWhiteMugOnLeftPlateAndPutYellowAndWhiteMugOnRightPlate
9L10L6PutWhiteMugOnPlateAndPutChocolatePuddingToRightPlate
10L10S1PickUpTheBookAndPlaceItInTheBackCompartmentOfTheCaddy
11L90K10CloseTheTopDrawerOfTheCabinet
12L90K10CloseTheTopDrawerOfTheCabinetAndPutTheBlackBowlOnTopOfIt
13L90K10PutTheBlackBowlInTheTopDrawerOfTheCabinet
14L90K10PutTheButterAtTheBackInTheTopDrawerOfTheCabinetAndCloseIt
15L90K10PutTheButterAtTheFrontInTheTopDrawerOfTheCabinetAndCloseIt
16L90K10PutTheChocolatePuddingInTheTopDrawerOfTheCabinetAndCloseIt
17L90K1OpenTheBottomDrawerOfTheCabinet
18L90K1OpenTheTopDrawerOfTheCabinet
19L90K1OpenTheTopDrawerOfTheCabinetAndPutTheBowlInIt
20L90K1PutTheBlackBowlOnThePlate
21L90K1PutTheBlackBowlOnTopOfTheCabinet
22L90K2OpenTheTopDrawerOfTheCabinet
23L90K2PutTheBlackBowlAtTheBackOnThePlate
24L90K2PutTheBlackBowlAtTheFrontOnThePlate
25L90K2PutTheMiddleBlackBowlOnThePlate
26L90K2PutTheMiddleBlackBowlOnTopOfTheCabinet
27L90K2StackTheBlackBowlAtTheFrontOnTheBlackBowlInTheMiddle
28L90K2StackTheMiddleBlackBowlOnTheBackBlackBowl
29L90K3PutTheFryingPanOnTheStove
30L90K3PutTheMokaPotOnTheStove
31L90K3TurnOnTheStove
32L90K3TurnOnTheStoveAndPutTheFryingPanOnIt
33L90K4CloseTheBottomDrawerOfTheCabinet
34L90K4CloseTheBottomDrawerOfTheCabinetAndOpenTheTopDrawer
35L90K4PutTheBlackBowlInTheBottomDrawerOfTheCabinet
36L90K4PutTheBlackBowlOnTopOfTheCabinet
37L90K4PutTheWineBottleInTheBottomDrawerOfTheCabinet
38L90K4PutTheWineBottleOnTheWineRack
39L90K5CloseTheTopDrawerOfTheCabinet
40L90K5PutTheBlackBowlInTheTopDrawerOfTheCabinet
41L90K5PutTheBlackBowlOnThePlate
42L90K5PutTheBlackBowlOnTopOfTheCabinet
43L90K5PutTheKetchupInTheTopDrawerOfTheCabinet
44L90K6CloseTheMicrowave
45L90K6PutTheYellowAndWhiteMugToTheFrontOfTheWhiteMug
46L90K7OpenTheMicrowave
47L90K7PutTheWhiteBowlOnThePlate
48L90K7PutTheWhiteBowlToTheRightOfThePlate
49L90K8PutTheRightMokaPotOnTheStove
50L90K8TurnOffTheStove
51L90K9PutTheFryingPanOnTheCabinetShelf
52L90K9PutTheFryingPanOnTopOfTheCabinet
53L90K9PutTheFryingPanUnderTheCabinetShelf
54L90K9PutTheWhiteBowlOnTopOfTheCabinet
55L90K9TurnOnTheStove
56L90K9TurnOnTheStoveAndPutTheFryingPanOnIt
57L90L1PickUpTheAlphabetSoupAndPutItInTheBasket
58L90L1PickUpTheCreamCheeseBoxAndPutItInTheBasket
59L90L1PickUpTheKetchupAndPutItInTheBasket
60L90L1PickUpTheTomatoSauceAndPutItInTheBasket
61L90L2PickUpTheAlphabetSoupAndPutItInTheBasket
62L90L2PickUpTheButterAndPutItInTheBasket
63L90L2PickUpTheMilkAndPutItInTheBasket
64L90L2PickUpTheOrangeJuiceAndPutItInTheBasket
65L90L2PickUpTheTomatoSauceAndPutItInTheBasket
66L90L3PickUpTheAlphabetSoupAndPutItInTheTray
67L90L3PickUpTheButterAndPutItInTheTray
68L90L3PickUpTheCreamCheeseAndPutItInTheTray
69L90L3PickUpTheKetchupAndPutItInTheTray
70L90L3PickUpTheTomatoSauceAndPutItInTheTray
71L90L4PickUpTheBlackBowlOnTheLeftAndPutItInTheTray
72L90L4PickUpTheChocolatePuddingAndPutItInTheTray
73L90L4PickUpTheSaladDressingAndPutItInTheTray
74L90L4StackTheLeftBowlOnTheRightBowlAndPlaceThemInTheTray
75L90L4StackTheRightBowlOnTheLeftBowlAndPlaceThemInTheTray
76L90L5PutTheRedMugOnTheLeftPlate
77L90L5PutTheRedMugOnTheRightPlate
78L90L5PutTheWhiteMugOnTheLeftPlate
79L90L5PutTheYellowAndWhiteMugOnTheRightPlate
80L90L6PutTheChocolatePuddingToTheLeftOfThePlate
81L90L6PutTheChocolatePuddingToTheRightOfThePlate
82L90L6PutTheRedMugOnThePlate
83L90L6PutTheWhiteMugOnThePlate
84L90S1PickUpTheBookAndPlaceItInTheFrontCompartmentOfTheCaddy
85L90S1PickUpTheBookAndPlaceItInTheLeftCompartmentOfTheCaddy
86L90S1PickUpTheBookAndPlaceItInTheRightCompartmentOfTheCaddy
87L90S1PickUpTheYellowAndWhiteMugAndPlaceItToTheRightOfTheCaddy
88L90S2PickUpTheBookAndPlaceItInTheBackCompartmentOfTheCaddy
89L90S2PickUpTheBookAndPlaceItInTheFrontCompartmentOfTheCaddy
90L90S2PickUpTheBookAndPlaceItInTheLeftCompartmentOfTheCaddy
91L90S2PickUpTheBookAndPlaceItInTheRightCompartmentOfTheCaddy
92L90S3PickUpTheBookAndPlaceItInTheFrontCompartmentOfTheCaddy
93L90S3PickUpTheBookAndPlaceItInTheLeftCompartmentOfTheCaddy
94L90S3PickUpTheBookAndPlaceItInTheRightCompartmentOfTheCaddy
95L90S3PickUpTheRedMugAndPlaceItToTheRightOfTheCaddy
96L90S3PickUpTheWhiteMugAndPlaceItToTheRightOfTheCaddy
97L90S4PickUpTheBookInTheMiddleAndPlaceItOnTheCabinetShelf
98L90S4PickUpTheBookOnTheLeftAndPlaceItOnTopOfTheShelf
99L90S4PickUpTheBookOnTheRightAndPlaceItOnTheCabinetShelf
100L90S4PickUpTheBookOnTheRightAndPlaceItUnderTheCabinetShelf
101LGOpenTheMiddleDrawerOfTheCabinet
102LGOpenTheTopDrawerAndPutTheBowlInside
103LGPushThePlateToTheFrontOfTheStove
104LGPutTheBowlOnThePlate
105LGPutTheBowlOnTheStove
106LGPutTheBowlOnTopOfTheCabinet
107LGPutTheCreamCheeseInTheBowl
108LGPutTheWineBottleOnTheRack
109LGPutTheWineBottleOnTopOfTheCabinet
110LGTurnOnTheStove
111LOPickUpTheAlphabetSoupAndPlaceItInTheBasket
112LOPickUpTheBbqSauceAndPlaceItInTheBasket
113LOPickUpTheButterAndPlaceItInTheBasket
114LOPickUpTheChocolatePuddingAndPlaceItInTheBasket
115LOPickUpTheKetchupAndPlaceItInTheBasket
116LOPickUpTheMilkAndPlaceItInTheBasket
117LOPickUpTheOrangeJuiceAndPlaceItInTheBasket
118LOPickUpTheSaladDressingAndPlaceItInTheBasket
119LOPickUpTheTomatoSauceAndPlaceItInTheBasket
120LOPutCreamCheeseInBasket
121LSPickUpBlackBowlBetweenPlateAndRamekinAndPlaceItOnPlate
122LSPickUpBlackBowlInTopDrawerOfWoodenCabinetAndPlaceItOnPlate
123LSPickUpTheBlackBowlFromTableCenterAndPlaceItOnThePlate
124LSPickUpTheBlackBowlNextToTheCookieBoxAndPlaceItOnThePlate
125LSPickUpTheBlackBowlNextToThePlateAndPlaceItOnThePlate
126LSPickUpTheBlackBowlNextToTheRamekinAndPlaceItOnThePlate
127LSPickUpTheBlackBowlOnTheCookieBoxAndPlaceItOnThePlate
128LSPickUpTheBlackBowlOnTheRamekinAndPlaceItOnThePlate
129LSPickUpTheBlackBowlOnTheStoveAndPlaceItOnThePlate
130LSPickUpTheBlackBowlOnTheWoodenCabinetAndPlaceItOnThePlate

Data Structure & Access

Dataset structure example (from HuggingFace LightwheelAI/lightwheel_tasks):

lightwheel_tasks/
├── lightwheel_robocasa_tasks_g1_wbc/
│ └── BeverageOrganization/
│ ├── BeverageOrganization_1763692193809507/
│ │ ├── isaac_replay_state_product.mp4 # Robot demonstration video
│ │ ├── running_args.json # Collection parameters and environment metadata
│ │ └── trajectories.hdf5 # Trajectory data (multi-modal: state, actions, joint targets etc.)
│ └── ...
└── lightwheel_libero_tasks_piper/
└── L90K3TurnOnTheStove/
├── L90K3TurnOnTheStove_1758698193690141/
│ ├── isaac_replay_state.mp4
│ ├── running_args.json
│ └── trajectories.hdf5
└── ...
  • Each (task/robot/episode) subfolder contains one robot demonstration video (mp4), a JSON file of parameters and metadata, and an HDF5 trajectory with full sequence observations.
  • File structure and naming are highly standardized for easy task filtering and programmatic loading.

Loading Example:

from datasets import load_dataset

ds = load_dataset("LightwheelAI/lightwheel_tasks")
print(ds)

Applications

  • Imitation learning
  • Reinforcement learning
  • Generalization, transfer, and multi-robot evaluation
  • High-fidelity kitchen manipulation research

Citation

If you use this dataset, please cite and reference Lightwheel Kitchen Manipulation Task Dataset.