Data processing

[neerjathakkar]

I am wondering if the collect_data() function in social_utils.py was used to create the .pickle files? When I try to use that function to process a different dataset the output shapes of the trajectories don't make sense for the dataloader, but it is unclear how to modify the function. It would also be helpful to access the original ETH and SDD dataset files to understand how the data was pre-processed.