【Argoverse 1 Motion Forecasting Dataset】轨迹预测数据集使用与评价指标-Toy模板网

这篇具有很好参考价值的文章主要介绍了【Argoverse 1 Motion Forecasting Dataset】轨迹预测数据集使用与评价指标。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

1.数据集数据分析

Argoverse 1https://www.argoverse.org/av1.html#forecasting-link下载对应的“Argoverse HD Maps”以及“Argoverse Motion Forecasting v1.1”

argoverse数据集,轨迹预测数据集与评价指标,人工智能

1.1数据集分析

通过下载Sample Datasets v1.1>Motion Forecasting文件包对数据集进行分析。这里给出了5个场景的数据csv文件，每个csv文件代表一个场景，读取任一文件，主要包括“时间戳TIMESTAMP、跟踪id TRACK_ID、目标类别OBJECT_TYPE、坐标x,y、采集的城市CITY_NAME"

argoverse数据集,轨迹预测数据集与评价指标,人工智能

1.1.1 时间戳 TIMESTAMP

采集频率是10Hz，因此时间步是0.1s，总长是5s，意味着每个场景每个目标最多有50个样本点；其中AV、AGENT样本数固定为50个，OTHERS最多为50个(OTHTER在场景中有出现和消失地现象)。

1.1.2 跟踪id TRACK_ID

作为识别和匹配的依据。

1.1.3 目标类别OBJECT_TYPE

主要包括：自动驾驶自车AV、显示感兴趣的车辆AGENT、其他感兴趣的对象OTHERS.

1.1.4 坐标x,y

示例中给出了三种类别的坐标精确到小数点后7位。是连续轨迹的离散化。

argoverse数据集,轨迹预测数据集与评价指标,人工智能

Argovers数据集目标检测部分给出的类别挺多的，但是在轨迹预测上并没有给出类别标签，这是限制了预测模型性能的重要原因之一，因为同种场景下，同种类别不同类型智能体对其他智能体的交互影响是显式地不一样的。

1.1.5采集的城市CITY_NAME

匹配高精地图。

2.HD Map API 的使用

GitHub - argoverse/argoverse-api: Official GitHub repository for Argoverse datasethttps://github.com/argoverse/argoverse-api

2.1 API 的安装

2.1.1 Clone

在创建的项目环境下（是conda创建的项目环境），通过复制以下 repo 下载API：

git clone https://github.com/argoai/argoverse-api.git

2.2.2 Download HD map data

从Argoverse 1下载 hd_maps.tar.gz 并解压缩到repo的根目录中。目录结构应该如下所示：

argodataset
└── argoverse
└── data_loading
└── evaluation
└── map_representation
└── utils
└── visualization
└── map_files
└── license
...

2.2.3 Download Argoverse-Forecasting

Argoverse 提供了完整的数据集和数据集的样本版本，用于测试目的。Argorverse预测包含327790个有趣的场景序列。每个序列都跟随主要智能体的轨迹5秒，同时跟踪所有其他参与者（如汽车、行人）。我们将它们分为208272个训练序列、40127个验证序列和79391个测试序列。我们还在forecasting_sample.tar.gz中提供了样本数据（5个序列）。

需要注意的是，需要下载高清地图数据（并将其提取到项目根文件夹中），以便API正常运行。下载Argorvers Forecasting，数据可以提取到本地计算机中的任何位置。

2.2.4 Install argoverse package

argoverse可以作为python包安装，使用

  pip install -e /path_to_root_directory_of_the_repo/

确保可以在python中运行 python -c “import-argoverse”，就可以开始了！

2.2 使用

Argorverse API提供了与我们数据集的三个主要组件交互的有用功能：HD地图、Argorvers跟踪数据集和Argorverses预测数据集。

from argoverse.map_representation.map_api import ArgoverseMap
from argoverse.data_loading.argoverse_forecasting_loader import ArgoverseForecastingLoader
avm = ArgoverseMap()
argoverse_forecasting_loader = ArgoverseForecastingLoader('argoverse-forecasting/') #simply change to your local path of the data

此处提供API文档。建议通过以下演示教程开始学习。

2.3 Demo

展示如何使用API 在Jupyter Notebook中提供了使用的Demo.

首先，下载安装jupyter notebook : pip install jupyter. 在repo目录下打开Jupyter notebook，也可以在根目录中一层一层寻找（比较麻烦）。

运行后，只需导航到 demo_usage 文件夹并打开任何教程！要使用预测教程，需要从网站下载预测样本数据，并将文件夹提取到repo的根目录中。

2.3.1 Argovese地图教程

argoverse数据集,轨迹预测数据集与评价指标,人工智能

2.3.2 Argorverse预测教程

argoverse数据集,轨迹预测数据集与评价指标,人工智能

2.3.3 渲染鸟瞰图

运行以下脚本以从地图上的鸟瞰图渲染长方体。

$ python visualize_30hz_benchmark_data_on_map.py --dataset_dir <path/to/logs> --log_id <id of the specific log> --experiment_prefix <prefix of the output directory>

例如，日志的路径可能是argoverse tracking/train4，日志id可能是2bc6a872-9979-3493-82eb-fb55407473c9。此脚本将使用如下图像写入当前工作目录中的<experienceprefix>_per_log_viz/<log-id>：

它还将在 <experienceprefix>_per_log_viz/<log-id>_lidar_roi_nonground.mp4 生成视频可视化

3.预测baseline

GitHub - jagjeet-singh/argoverse-forecasting: Official Repository for Argoverse Motion Forecasting BaselinesOfficial Repository for Argoverse Motion Forecasting Baselines - GitHub - jagjeet-singh/argoverse-forecasting: Official Repository for Argoverse Motion Forecasting Baselineshttps://github.com/jagjeet-singh/argoverse-forecasting文章来源地址https://www.toymoban.com/news/detail-808947.html

到了这里，关于【Argoverse 1 Motion Forecasting Dataset】轨迹预测数据集使用与评价指标的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！