search

ARISE-Initiative / robomimic

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

Cannot download the robomimic dataset with image #123

Closed StarCycle closed 10 months ago

StarCycle commented 10 months ago

Hello,

I cannot download the robomimic dataset with the following url:

http://downloads.cs.stanford.edu/downloads/rt_benchmark/lift/mg/image_sparse_v141.hdf5 http://downloads.cs.stanford.edu/downloads/rt_benchmark/lift/mg/image_dense_v141.hdf5 http://downloads.cs.stanford.edu/downloads/rt_benchmark/can/mg/image_sparse_v141.hdf5 http://downloads.cs.stanford.edu/downloads/rt_benchmark/can/mg/image_dense_v141.hdf5

I get these url from the DATASET_REGISTRY dict in your colab script (https://colab.research.google.com/drive/1b62r_km9pP40fKF0cBdpdTO2P_2eIbC6?usp=sharing#scrollTo=cfsnmbDYv7xZ)

Do you change the address of the dataset? Or there is a newer version?

I am looking for a dataset with images in simulated environment, so I can evaluate the policy in your simulator without sim2real.

Kind regards, Zhuoheng

amandlek commented 10 months ago

Thanks for pointing this out! We don't actually have direct download links for these datasets that are based on robosuite v1.4.1 (see note here about this change). Instead, we recommend that you extract image observations from the demo_v141.hdf5 datasets. See here for more info on this, and here for example commands you can use.

We will change these links in the DATASET_REGISTRY to avoid confusion (tagging @snasiriany to handle this).

Another option is to use robomimic v0.2 and the download links there, along with robosuite v1.2 on the offline_study branch (as noted here).

snasiriany commented 10 months ago

I started a PR to fix this confusion: https://github.com/ARISE-Initiative/robomimic/pull/127