realcrane / Human-Trajectory-Prediction-via-Neural-Social-Physics

Our ECCV 2022 paper Human Trajectory Prediction via Neural Social Physics
127 stars 19 forks source link

'train.pickle' and 'test.pickle' file in the SDD folder #25

Open jjiwon-kim opened 10 months ago

jjiwon-kim commented 10 months ago

Hi! First, thank you for your great work. It's been a great help for my research.

I have a simple question regarding two files in the data/SDD directory, which are 'train.pickle' and 'test.pickle'. I found that they have different formats from the pickle files generated directly from the actual SDD datasets using your preprocessing codes (processing_SDD.py and generate_train_test_data_SDD.py in SDD_ini folder). I found out this issue while trying to use my own data to test and train the 'train_goals.py'. In order to do so, I first prepared my data in the same format as SDD dataset, and then used your preprocessing codes to turn them into pickle files. Then I put those into 'train_goals.py' only to find out that they are specifically coded to get 'train.pickle' and 'test.pickle' (in the data/SDD folder) as input.

So, I wonder 1. what is the difference between the provided pickle files and those that are generated using the preprocessing codes, and 2. how I can make my own SDD-like data into the usable format for 'train_goals.py'.