CoolFace
Datasetpublic

pravsels/arx5-robot-dataset

ARX5 Manipulation Dataset Robotic manipulation demos collected on a single ARX5 arm, of tabletop tasks with dual RGB cameras and DINOv2 embeddings of the images. Converted from Ville Kuosmanen's LeRobot v2.1 datasets into a unified HDF5 format. Hardware Robot: ARX5 single arm (7-DoF) Cameras: Wrist-mounted + front-facing scene camera Stats Trajectories 1,704 Total frames 860,076 Frames per trajectory 4 – 3,034 (median 397) Total… See the full description on the dataset page: https://huggingface.co/datasets/pravsels/arx5-robot-dataset.

sourceHugging Faceapache-2.0updated 10mo agoView on Hugging Face
2likes470downloads
Dataset Card

ARX5 Manipulation Dataset

Robotic manipulation demos collected on a single ARX5 arm, of tabletop tasks with dual RGB cameras and DINOv2 embeddings of the images.

Converted from Ville Kuosmanen's LeRobot v2.1 datasets into a unified HDF5 format.

Hardware

  • Robot: ARX5 single arm (7-DoF)
  • Cameras: Wrist-mounted + front-facing scene camera

Stats

Trajectories1,704
Total frames860,076
Frames per trajectory4 – 3,034 (median 397)
Total size1 TB+

Quick Start

python
import h5py

with h5py.File("arx5_datasets_part_001.h5", "r") as f:
    traj = f["trajectory_0042"]
    actions = traj["actions"][:]
    images = traj["camera_0"][:]

The dataset is split across 18 HDF5 files (100 trajectories each), from arx5_datasets_part_001.h5 to arx5_datasets_part_018.h5

Each part contains trajectories renumbered trajectory_0000 through trajectory_0099 within the chunk.

What's Inside

Tasks: Picking coffee props, dropping a footbag into a dice tower, rice cooker assembly (multi-step), plus DAgger correction runs.

Per frame:

  • actions / states — 7-DoF joint commands and observations
  • camera_0 / camera_1 — 480×640 RGB (wrist / scene)
  • cam_rs_embd / cam_zed_embd — DINOv2 patch embeddings (256×384)

Source Datasets

This dataset aggregates the following HuggingFace datasets:

DatasetTrajectories
villekuosmanen/dAggercoffeeprop128
villekuosmanen/rice-cookerstep2100
villekuosmanen/rice-cookerstep3100
villekuosmanen/dropfootbagintodicetower100
villekuosmanen/rice-cookerstep099
villekuosmanen/pickcoffeeprop_20Hz99
villekuosmanen/dAggerplusv2coffeeprop60
villekuosmanen/dAggercoffeeprop_2.1.040
villekuosmanen/dAggercoffeeprop_2.2.040
villekuosmanen/dAggercoffeeprop_2.3.040
villekuosmanen/dAggercoffeeprop_2.4.040
villekuosmanen/dAggercoffeeprop_3.1.140
villekuosmanen/dAggerdropfootbagintodicetower1.4.040
villekuosmanen/dAgger_test40
villekuosmanen/faildropfootbagintodice_tower40
villekuosmanen/dAggercoffeeprop_3.0.039
villekuosmanen/evaldAggerdropfootbagintodicetower_1.7.037
villekuosmanen/eval_26Jan2536
villekuosmanen/eval_27Jan2536
villekuosmanen/eval2Feb25v236
villekuosmanen/eval_3Feb2536
villekuosmanen/pickcoffeefilters30
villekuosmanen/pickcoffeeprop20Hz330
villekuosmanen/eval_29Jan2530
villekuosmanen/eval31Jan25v230
villekuosmanen/eval_8Jun2529
villekuosmanen/eval_30May2527
villekuosmanen/eval8Jun25aug27
villekuosmanen/eval30May25alt24
villekuosmanen/eval_31May2524
villekuosmanen/eval_3Mar2524
villekuosmanen/eval_28May2523
villekuosmanen/eval_4Apr2523
villekuosmanen/eval_31Jan2522
villekuosmanen/dropfootbagintodicetower_continuous20
villekuosmanen/dAggerplustest20
villekuosmanen/dAggerpluscoffee_prop19
villekuosmanen/eval_6Mar2519
villekuosmanen/eval26Jun25412
villekuosmanen/evaldropthefootbagintothedicetower40a5ea4912
villekuosmanen/pickcoffeeprop20Hzalt12
villekuosmanen/evaldropthefootbagintothedicetower1afce3af10
villekuosmanen/evaldropthefootbagintothedicetower9bbb4c345
villekuosmanen/arx5-eef-pose-test2
villekuosmanen/arx5-vel-effort-test2
villekuosmanen/dAggerplusextra_test2

License

Apache 2.0