yunshengtian / ASAP

[ICRA 2024] ASAP: Automated Sequence Planning for Complex Robotic Assembly with Physical Feasibility
http://asap.csail.mit.edu
MIT License
40 stars 5 forks source link

Hello, can I ask for the planning sequence of the dataset. #7

Open HonsChon opened 1 month ago

HonsChon commented 1 month ago

Hello, can I ask for the planning sequence of the dataset. Thank you so much.

yunshengtian commented 3 weeks ago

Hi, thanks for your interest in the dataset! Unfortunately, I need some time to dig out the dataset that we used to train the network since I am traveling right now. But we released the pretrained model on that dataset, and also the dataset can be generated by our method fully automatically (say if you want to generate a larger dataset that's totally doable).

Generating the dataset can be straightforward since here is how you can get the label - basically the 'feasible' key in the returned dict is the label. In the meantime, I will try to find out the dataset that we used.