search

omron-sinicx / neural-astar

Official implementation of "Path Planning using Neural A* Search" (ICML-21)
https://omron-sinicx.github.io/neural-astar
Other
236 stars 57 forks source link

dataloader for sdd #23

Closed zwxu064 closed 1 year ago

zwxu064 commented 1 year ago

Hi, Do you have the dataloader for SDD? For CSM and TiledMP, there are a single .npz for each dataset; but for SDD, there are 8 types each type has some "video" subfolder where each subfolder has several .npz. So I am wondering if you have the dataloader for SDD for reproduce? Thanks.

yonetaniryo commented 1 year ago

Hi! Please refer to the icml2021 branch.

https://github.com/omron-sinicx/neural-astar/blob/ff7ef1684facb451662e942aaa4978432a1cbead/planning_experiment/utils/SDD.py#L12

https://github.com/omron-sinicx/neural-astar/blob/ff7ef1684facb451662e942aaa4978432a1cbead/scripts/train_sdd.py#L72

zwxu064 commented 1 year ago

Thank you very much.