search

google-research / ravens

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.
https://transporternets.github.io
Apache License 2.0
538 stars 93 forks source link

Dataset all frames #16

Closed Advaya24 closed 2 years ago

Advaya24 commented 2 years ago

Hi, is there a simple way to generate the dataset with all frames (including the intermediate observations while the action is being performed)? The current dataset only gives frames for before and after an action has been performed.

kevinzakka commented 2 years ago

Hey @Advaya24 , this should now be possible with the continuous arg. Check #11, it should have enough details!

Advaya24 commented 2 years ago

Oh, I see, thank you!