This repository is released for P2B in our CVPR 2020 paper (oral). Here we include our P2B model (PyTorch) and code for data preparation, training and testing on KITTI tracking dataset.
Install python 3.6
.
Install dependencies.
pip install -r requirements.txt
Build _ext
module.
python setup.py build_ext --inplace
Download the dataset from KITTI Tracking.
Download velodyne, calib and label_02 in the dataset and place them under the same parent folder.
Train a new P2B model on KITTI data:
python train_tracking.py --data_dir=<kitti data path>
Test a new P2B model on KITTI data:
python test_tracking.py --data_dir=<kitti data path>
Please refer to the code for setting of other optional arguments, including data split, training and testing parameters, etc.
Thank Giancola for his implementation of SC3D. Thank Erik Wijmans for his implementation of PointNet++ in PyTorch. Thank Charles R. Qi for his implementation of Votenet. They help and inspire this work.