Skip to content

Add dataset collection node for dora #1041

Merged
haixuanTao merged 10 commits intodora-rs:mainfrom
ShashwatPatil:dataset_record
Jul 16, 2025
Merged

Add dataset collection node for dora #1041
haixuanTao merged 10 commits intodora-rs:mainfrom
ShashwatPatil:dataset_record

Conversation

@ShashwatPatil
Copy link
Copy Markdown
Contributor

This node is able to record a dataset in LeRobot format.
It records joint angle, can record multiple camera feeds using the opencv-video-capture node.
Also added example to record dataset using SO101 arm

@haixuanTao
Copy link
Copy Markdown
Collaborator

Very cool!

Would it be possible to add within the example folder a quick guide on how to use it and what training would most likely look lite!

@ShashwatPatil
Copy link
Copy Markdown
Contributor Author

Yes, i will update the readme with usage instructions and the example

@haixuanTao
Copy link
Copy Markdown
Collaborator

FYI, I have added a fix on import naming based on latest main. I have also made it possible to train models without git cloning aerobat.

Would it be possible to add 3d visualisation of the arm using rerun in both dataflow?

@haixuanTao
Copy link
Copy Markdown
Collaborator

Would also be possible to remove the threading loop within the inference pipeline to make the prediction more simple.

Can we use a Lead Camera as the event message to use for running the inference.

@ShashwatPatil
Copy link
Copy Markdown
Contributor Author

Sure, will add the 3d visualization using rerun, and remove threading from the inference node

@ShashwatPatil
Copy link
Copy Markdown
Contributor Author

ShashwatPatil commented Jul 9, 2025

removed threading from inference node, also added optional saving of avif frames along with the mp4 videos that lerobot saves

@ShashwatPatil
Copy link
Copy Markdown
Contributor Author

Thanks for correcting the Python version 👍

@haixuanTao haixuanTao merged commit 758b090 into dora-rs:main Jul 16, 2025
153 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants