so100-synth-dataset / README.md
pranavsaroha's picture
Upload README.md with huggingface_hub
2adb037 verified
# So100 Synthetic Robot Dataset
This dataset contains synthetic robot pose estimation data for the LeRobot So100 arm.
## Dataset Structure
```
synth_dataset/
β”œβ”€β”€ data/
β”‚ └── chunk-000/
β”‚ └── episode_000000.parquet # Robot joint positions and metadata
β”œβ”€β”€ videos/
β”‚ └── chunk-000/
β”‚ └── observation.images.main/
β”‚ └── episode_000000.mp4 # RGB video frames
└── meta/
β”œβ”€β”€ episodes_stats.jsonl # Episode statistics
β”œβ”€β”€ episodes.jsonl # Episode metadata
β”œβ”€β”€ info.json # Dataset information
└── tasks.jsonl # Task definitions
```
## Dataset Information
- **Robot Type**: LeRobot So100 arm
- **Number of Frames**: 348
- **Video Format**: MP4 (24 FPS)
- **Data Format**: Parquet files containing joint positions and metadata
- **Source**: Synthetic video with inferred motor positions from trained pose estimation model
## Usage
This dataset can be used for:
- Robot pose estimation training and evaluation
- Joint angle prediction from RGB images
- Synthetic-to-real transfer learning
- Robot manipulation research
## Generation Process
1. **Input**: Synthetic RGB video of So100 arm movements
2. **Model**: Trained HoRoPose model (DepthNet + Full Network)
3. **Inference**: Joint angle predictions for each frame
4. **Dataset Creation**: Combined video frames with predicted joint positions
## Citation
If you use this dataset in your research, please cite the original HoRoPose paper and this dataset.
## License
[Add your license information here]