核心概念
This paper introduces a standardized toolbox for preprocessing and evaluating trajectory prediction models on the Drone datasets (highD, rounD, and inD), aiming to simplify comparative analysis and accelerate research in this field.
摘要
The paper highlights the need for standardizing the use of certain datasets for motion forecasting research to simplify comparative analysis. It proposes a set of tools and practices in the form of an open-sourced toolbox designed for researchers working on trajectory prediction problems.
The key aspects of the toolbox include:
-
Standardized preprocessing pipeline:
- Coordinate system transformation and downsampling
- Agent feature extraction and agent class definitions
- Semantic map construction using Lanelet2 information
- Dataset partitioning with stratified sampling
-
Evaluation metrics:
- Commonly used metrics like Average Displacement Error (ADE), Final Displacement Error (FDE), Average Path Displacement Error (APDE), Miss Rate (MR), Collision Rate (CR), Brier-FDE, and Average Negative Log-Likelihood (ANLL)
- Metrics consider both point-based and distribution-based predictions
-
Modular and extensible design:
- Implemented using PyTorch, PyTorch Geometric, and PyTorch Lightning
- Allows users to easily modify and extend the toolbox to suit their specific needs
The toolbox aims to alleviate the technical burden of developing research pipelines, thereby accelerating discoveries in the trajectory prediction field. It also opens up several future research directions, such as investigating scenario-specific performance, model generalization and zero-shot learning, and transfer learning across datasets.