How to run the model on a custom dataset?

cshizhe / robot_sugar

Official implementation of "SUGAR: Pre-training 3D Visual Representations for Robotics" (CVPR'24).

27 stars 0 forks source link

How to run the model on a custom dataset? #1

Open chanhee-luke opened 2 months ago

chanhee-luke commented 2 months ago

Hello! I stopped by your poster at CVPR and glad to see that the code is released! I want to run your model on a custom dataset for robot manipulation where input is a point cloud and output is a text or a bounding box. Can you guide me on where to start?

Thank you!

-Luke

cshizhe commented 2 months ago

Hi, thanks for your interest in our work. You could check the rlbench downstream task for your project.