search

ARISE-Initiative / robomimic

robomimic: A Modular Framework for Robot Learning from Demonstration
MIT License
595 stars 182 forks source link

Questions about dataset_states_to_obs.py #98

Closed WangYixuan12 closed 11 months ago

WangYixuan12 commented 11 months ago

Hi! Thank you all for the fantastic work! I am trying to look for one way to extract observations from the downloaded demo.hdf5. I found the instruction in the tutorial. I liked the idea of it. However, I don't know how to extract depth observation from it. It seems that I can only extract RGB observations only. Do you have any ideas on how to extract depth observation from the demo.hdf5?

amandlek commented 11 months ago

If you're talking about robosuite datasets, that would require a few changes to the script and the EnvRobosuite class. It could also require some other changes in the codebase to make sure depth images are processed correctly for training. We could put it on our roadmap for the next release if there's interest.

WangYixuan12 commented 11 months ago

Actually, I have found one way of rendering depth images. If you are interested, I could create a pull request and discuss it. If that has lower priority, we could close this issue.

snasiriany commented 11 months ago

Please share the pull request with us!

WangYixuan12 commented 11 months ago

Thanks! I just created the pull request #99. We may discuss there.