simtrack-MindSpore

Implemented the simtrack model based on MindSpore

Exploring Simple 3D Multi-Object Tracking for Autonomous Driving

[Paper]

requirements

Ubuntu 20.04
python=3.8
MindSpore=1.8
cuda=11.1

Getting Started

Data Preparation

• nuScenes

python ./datasets/create_data.py nuscenes_data_prep --root_path=NUSCENES_TRAINVAL_DATASET_ROOT --version="v1.0-trainval" --nsweeps=10

Training

python -m torch.distributed.launch --nproc_per_node=8 ./train.py examples/point_pillars/configs/nusc_all_pp_centernet_tracking.py --work_dir SAVE_DIR

Test

In ./model_zoo we provide our trained (pillar based) model on nuScenes.
Note: We currently only support inference with a single GPU.

python ./val_nusc_tracking.py examples/point_pillars/configs/nusc_all_pp_centernet_tracking.py --checkpoint CHECKPOINTFILE  --work_dir SAVE_DIR