Add dataset collection node for dora #1041
Conversation
|
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! |
|
Yes, i will update the readme with usage instructions and the example |
|
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? |
|
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. |
|
Sure, will add the 3d visualization using rerun, and remove threading from the inference node |
|
removed threading from inference node, also added optional saving of avif frames along with the mp4 videos that lerobot saves |
|
Thanks for correcting the Python version 👍 |
This node is able to record a dataset in LeRobot format.
It records joint angle, can record multiple camera feeds using the
opencv-video-capturenode.Also added example to record dataset using SO101 arm