NVlabs / trajdata

A unified interface to many trajectory forecasting datasets.
Apache License 2.0
290 stars 38 forks source link

Questions regarding Scene Agent Aggregation #23

Closed jxmmy7777 closed 1 year ago

jxmmy7777 commented 1 year ago

Hi, I would like to know how I can aggregate all the agents in the same scene into the same batch using trajdata. Is there a code in the repo that I can modify it to achieve this? I want to gather information from all agents simultaneously for further analysis.

In addition, I would also like to ask about normalizing the states using the dataset statistics.

Thank you for your attention!

jxmmy7777 commented 1 year ago

Find the instructions in examples/scenetimebatcher_example.py ! Thanks