RLBench Training Data Packaged format

nickgkan / 3d_diffuser_actor

Code for the paper "3D Diffuser Actor: Policy Diffusion with 3D Scene Representations"

https://3d-diffuser-actor.github.io/

MIT License

159 stars 16 forks source link

RLBench Training Data Packaged format #20

Closed richielo closed 3 months ago

richielo commented 3 months ago

Hello, thanks for open-sourcing the code. I am trying to train the model with my own RLBench data. But I can't seem to locate the script to convert the data to the .dat format. Can you point me to the relevant script? I see there is this script from one of the linked projects: https://github.com/zhouxian/act3d-chained-diffuser/tree/main/data_preprocessing. But I am not sure if that is the correct one. Thanks a lot!

twke18 commented 3 months ago

Hi,

We upload the python scripts for re-rendering high-resolution camera views and packaging episodes into .dat files. You can check DATA_PREPARATION_RLBENCH.md for details.

richielo commented 3 months ago

Thank you for uploading the conversion scripts!