stepjam / RLBench

A large-scale benchmark and learning environment.
https://sites.google.com/corp/view/rlbench
Other
1.03k stars 218 forks source link

[Feature Request] Can the gym environment provide depth images and point clouds? #198

Open StarCycle opened 9 months ago

StarCycle commented 9 months ago

https://github.com/stepjam/RLBench/blob/46cb9d97003961428f491dd7cc75433f6ab351dd/rlbench/gym/rlbench_env.py#L50-L63

Although the original RLBench environment includes depth images and point clouds in the observation, the gym version does not provide such information. They are important if researchers want to render images from new viewpoints (e.g., Robotic View Transformer). Could you please add them in the code?

I can modify the code and pull a request if you like it.

Kind regards, Zhuoheng