JunweiLiang / Multiverse

Dataset, code and model for the CVPR'20 paper "The Garden of Forking Paths: Towards Multi-Future Trajectory Prediction". And for the ECCV'20 SimAug paper.

Home Page:https://next.cs.cmu.edu/multiverse/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Feeding more features into forking path dataset

Jacobieee opened this issue · comments

Hi, I currently want to use the bounding box extracted in pickles files and feed them into the model when doing multifuture inference in multifuture_inference.py and I am not sure I understand the preprocess correctly. I noticed that in Virat preprocess, you stored every other box coordinates for each person in each frame. Thus, for data["other_box"] in each data sample, there is an array of lengths 8 which contains coordinates of other boxes at each time.

But for the forking path dataset, I only need to process the other box feature for the "controlled agent", right? So, the final processed other_box will be [N, obs_len, K] where K is the number of other boxes and N is 507, the number of the testing set.

Thanks for that.

The statement about VIRAT preprocess is correct. For the forking path dataset, there are 507 test samples so yes the other_box should be [507, obs_len, K, 4]