Farama-Foundation / Gymnasium-Robotics

A collection of robotics simulation environments for reinforcement learning
https://robotics.farama.org/
MIT License
571 stars 91 forks source link

[Proposal] FrankaKitchen RGB base observation ? #252

Closed Lr-2002 closed 1 month ago

Lr-2002 commented 1 month ago

Proposal

Could the Frankakitchen Env have a RGB Mode ?

Motivation

I'm now doing some research on whether is it possible to use world model to solve Visual Long-term task, as the Franka Kitchen is an awesome env for robotics manipulation, so I'm now very interested in whether could the Franka-kitchen support vision mode observation ?

Pitch

env.step() could return the obs who is the ndarray standing for the image

Alternatives

I'm now could only think that the render function could give some image back, but I'm not sure the computation efficiency of it.

Checklist

Kallinteris-Andreas commented 1 month ago

https://gymnasium.farama.org/api/wrappers/observation_wrappers/#gymnasium.wrappers.AddRenderObservation

Lr-2002 commented 1 month ago

wow, sorry for i have no idea about the gymnasium has the wrapper already

Lr-2002 commented 1 month ago

Thanks for your quick response ! Best wishes