SaikiranKannaiah436 / argoverse_baselinetracker

Baseline tracker code release for the paper Argoverse: 3D Tracking and Forecasting With Rich Maps, CVPR 2019.

Argoverse Baseline Tracker

Baseline tracker code release for the paper Argoverse: 3D Tracking and Forecasting With Rich Maps, CVPR 2019. [paper]  [website] 

Introduction

This tracker implementation is meant to be a baseline example to demonstrate the use of the map and tracking data in Argoverse dataset. We don't claim its performance to be the best, and we are looking forward to more map-based tracking methods developed using Argoverse in the future.

Requirements

To run this tracker, please install following requirements:

• argoverse api
• sklearn
• pyquaternion
• uuid
• mayavi (for 3D visualization)
• MaskRCNN (commit 55796a04ea770029a80cf5933cc5c3f3f6fa59cf) Follow the official guide to install
• pcl 1.8 and python-pcl binding
• Download smallestenclosingcircle.py
• Download MinimumBoundingBox
• Download Argoverse Tracking dataset from Argoverse official website.

example:

Command for running this tracker tracker on Argoverse:

python3 run_tracking.py --path_dataset=/path/to/argoverse-tracking/test --log_id=<log_id>  --path_output=/path/to/output/folder --use_maskrcnn --region_type=roi --use_map_lane --motion_model='const_v' --measurement_model=both --fix_bbox_size --dataset_name=Argoverse

You can use --save_bev_imgs to print birds-eye-view image as above example or --show_segmentation to plot 3D visualization of segmentation result if mayavi is installed. Tracker output format is the same as Argoverse tracking label format.

Docker Image

It might be tricky to install all the dependencies, so we provided docker image.

• Docker version 18.09.7, build 2d0083d

To run docker image, first install nvidia-docker and then run following command to build image using the provided DockerFile:

nvidia-docker build -t baselinetracker docker/

After building the image, run following command to start. Mount dataset folder so the data can be accessed in docker environment:

nvidia-docker run -v /path/to/argoverse-tracking/test:/data  -it baselinetracker:latest

And then clone this repo:

git clone https://github.com/alliecc/argoverse_baselinetracker
cd argoverse_baselinetracker
wget https://www.nayuki.io/res/smallest-enclosing-circle/smallestenclosingcircle.py

Here is an example command to start tracker. The tracking output would be stored in /tracking_output.

python3 run_tracking.py --path_dataset=/data --log_id=0f0d7759-fa6e-3296-b528-6c862d061bdd  --path_output=/tracking_output --use_maskrcnn --region_type=roi --use_map_lane --motion_model='const_v' --measurement_model=both --fix_bbox_size --dataset_name=Argoverse