CoolFace
Datasetpublic

THULab/Dangvi_soarm100_data2

SOARM100 Data2 TsFile Apache TsFile edition of Dangvi/soarm100_data2, a LeRobot v2.1 SO100 dataset with demonstrations for lifting orange or green 2x4 blocks above an 8x8 blue block. Source and attribution Original author, repository owner, and uploader: DANBI KIM (Dangvi). License: Apache-2.0. The source card provides no homepage, paper, or completed BibTeX citation. Split: train; 75 episodes; 33,440 frame rows; 2 tasks; 30 FPS; 75 source Parquet shards. The… See the full description on the dataset page: https://huggingface.co/datasets/THULab/Dangvi_soarm100_data2.

sourceHugging Faceapache-2.0updated 1mo agoView on Hugging Face
0likes37downloads
Dataset Card

SOARM100 Data2 TsFile

Apache TsFile edition of Dangvi/soarm100_data2, a LeRobot v2.1 SO100 dataset with demonstrations for lifting orange or green 2x4 blocks above an 8x8 blue block.

Source and attribution

  • —Original author, repository owner, and uploader: DANBI KIM (Dangvi).
  • —License: Apache-2.0.
  • —The source card provides no homepage, paper, or completed BibTeX citation.
  • —Split: train; 75 episodes; 33,440 frame rows; 2 tasks; 30 FPS; 75 source Parquet shards.
  • —The source card's embedded metadata example says 25 episodes, but the current meta/info.json, episode metadata, and repository file tree consistently contain 75 episodes.
  • —Task 0: Pick up the 2x4 orange block and go up the 8x8 blue block.
  • —Task 1: Pick up the 2x4 green block and go up the 8x8 blue block.

Data layout

The table is dangvi_soarm100_data2 and contains 33,440 rows across 75 episode/task devices.

ColumnTsFile roleTypeMeaning
TimeTIMEINT64 millisecondsround(timestamp * 1000), restarting at zero per episode
episode_indexTAGSTRING from source INT64Source episode identity
task_indexTAGSTRING from source INT64Source task identity
frame_indexFIELDINT64Frame position within the episode
sample_indexFIELDINT64Source index, renamed for clarity
action_0 ... action_5FIELDFLOATFlattened action[6]
observation_state_0 ... observation_state_5FIELDFLOATFlattened observation.state[6]

timestamp is not retained as a separate FIELD because it is represented by Time / 1000 seconds. Vector names preserve their source prefixes, with dots changed to underscores. No trajectory row, episode, task, action dimension, or state dimension is removed.

Videos and alignment

The 150 source AV1 videos remain in the original repository under `videos/chunk-000`:

The source template is videos/chunk-{episode_chunk:03d}/{video_key}/episode_{episode_index:06d}.mp4. Use episode_index and frame_index to align each numeric row with both 30 FPS video streams. The videos are not included in this TsFile dataset.

Read example

python
from tsfile import TsFileReader

reader = TsFileReader("data/dangvi_soarm100_data2.tsfile")
with reader.query_table(
    "dangvi_soarm100_data2",
    ["episode_index", "task_index", "frame_index", "sample_index", "action_0", "observation_state_0"],
    batch_size=1024,
) as result:
    batch = result.read_arrow_batch()
    print(batch.to_pandas().head())
reader.close()