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.
https://next.cs.cmu.edu/multiverse/
Apache License 2.0
250 stars 61 forks source link

Prepare training data? #38

Closed 108618026 closed 2 years ago

108618026 commented 2 years ago

Hello @JunweiLiang I would like to confirm that the data(sddactev_trainval/actev/) use to finetune SimAug model is the same as the outcomes of following your preprocessing rules in training??

  1. Get data split
  2. Get trajectories
  3. Get RGB frames and scene segmentation features
  4. Remove videos with bad trajectories
  5. Preprocess data into npz files for efficient training
python code/train.py sddactev_trainval/actev/ simaug_my_finetune actev_modelname \
 --load_from packed_models/best_simaug_model/00/best/ --wd 0.001 --runId 0 \
 --obs_len 8 --pred_len 12 --emb_size 32 --enc_hidden_size 256 --dec_hidden_size \
 256 --activation_func tanh --keep_prob 1.0 --num_epochs 30 --batch_size 20 \
 --init_lr 0.05 --use_gnn --learning_rate_decay 0.95 --num_epoch_per_decay 2.0 \
 --grid_loss_weight 1.0 --grid_reg_loss_weight 0.1 --save_period 1000 --scene_h 36 \
 --scene_w 64 --scene_conv_kernel 3 --scene_conv_dim 64 --scene_grid_strides 2,4 \
 --use_grids 1,0 --val_grid_num 0 --train_w_onehot --gpuid 0

Another question is that I would like to use the Argoverse dataset to finetune the SimAug model, whether it's possible to use three-view samples for training(e.g. front-center, front-left, front-right)??

Thanks in advance!

JunweiLiang commented 2 years ago

Yes the pre-processed data downloaded from the testing instructions should be the same from the preprocessing instructions. You can also check out this fork from the Google-Next-Prediction.

Another question is that I would like to use the Argoverse dataset to finetune the SimAug model, whether it's possible to use three-view samples for training(e.g. front-center, front-left, front-right)??

Yes! It should work as long as they have overlapping and synchronized views.