Is image observation of the Franka Kitchen dataset supported?

conglu1997 / v-d4rl

Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations

MIT License

94 stars 9 forks source link

Closed BrightMoonStar closed 7 months ago

BrightMoonStar commented 7 months ago

Hi~ I want to know whether image observation of the Franka Kitchen dataset supported? Thank you ！

conglu1997 commented 7 months ago

Hi! Yes, of course. You may need to change the size of the encoder, let me know if you need help!

BrightMoonStar commented 7 months ago

I'm very sorry that I couldn't find the relevant dataset of Franka Kitchen on hugging face or Google Drive, and there doesn't seem to be Franka Kitchen in the list of this previous reply. https://github.com/conglu1997/v-d4rl/issues/5#issuecomment-1732234639

conglu1997 commented 7 months ago

We have no Franka dataset. If you had your own Franka dataset, you would be able to use our algorithms with it!