erksch / 3D-object-tracking

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

3D Object Tracking on the Waymo Open Dataset

An approach to 3D Object Tracking which is evaluated on segments on the Waymo Open Dataset. The approach is heavily influenced by Weng's Baseline for 3D Multi-Object Tracking.

Setup

Create a virtual environment and install dependencies

python3 -m venv venv
source venv/bin/active
pip install -r requirements.txt

Usage

Download a dataset segment (.tfrecord file) and run the tracker evaluation with

python main.py \
  --tracker predictive \
  --segment path/to/segment.tfrecord \
  --dropout 0.2

About


Languages

Language:Python 100.0%