评测数据准备
简体中文 | English
1. 复用训练数据
完成对应环境的 SatNav 训练数据或 Habitat 训练数据准备后,以下资源可以直接用于评测:
环境 |
评测复用资源 |
评测不需要 |
|---|---|---|
SatNav |
SatNav-v0.1 |
|
Habitat |
R2R VLN-CE v1-3 |
R2R / RxR / EnvDrop 离线轨迹 |
此时只需在 ${SWIFTVLN_ROOT}/.local/env.sh 中补充评测路径:
# SatNav
export SWIFTVLN_SATNAV_EVAL_ROOT="${SWIFTVLN_SATNAV_DATA_ROOT}/${SWIFTVLN_SATNAV_DATASET}/episodes/eval"
export SWIFTVLN_SATNAV_EVAL_DATA_PATH="${SWIFTVLN_SATNAV_EVAL_ROOT}/{split}/all_episodes.json"
export SWIFTVLN_SATNAV_SCENES_DIR="${SWIFTVLN_SATNAV_DATA_ROOT}/scenes"
# Habitat
export SWIFTVLN_HABITAT_SCENES_DIR="${SWIFTVLN_HABITAT_DATA_ROOT}/scene_datasets"
export SWIFTVLN_HABITAT_R2R_EVAL_DATA_PATH="${SWIFTVLN_HABITAT_DATA_ROOT}/datasets/r2r/{split}/{split}.json.gz"
完成上述配置后,无需继续阅读本章,直接进入 SwiftVLN 评测。 以下章节面向仅准备评测数据的研究者。
3. Habitat 评测数据
3.1 准备 MP3D 场景
按照 Matterport3D 官方说明申请并下载 Habitat 场景, 将 90 个场景保存至:
/path/to/streamvln_datasets/scene_datasets/mp3d/
每个场景包含 Habitat 使用的 .glb 和 .navmesh:
scene_datasets/mp3d/
├── 17DRP5sb8fy/
│ ├── 17DRP5sb8fy.glb
│ └── 17DRP5sb8fy.navmesh
└── ...
3.2 下载 R2R VLN-CE v1-3
从 VLN-CE 数据页下载
R2R_VLNCE_v1-3.zip:
python -m pip install gdown
mkdir -p /path/to/streamvln_datasets/datasets
cd /path/to/streamvln_datasets/datasets
gdown --id 1T9SjqZWyR2PCLSXYkFckfDeIs6Un0Rjm
unzip R2R_VLNCE_v1-3.zip
mv R2R_VLNCE_v1-3 r2r
评测使用:
Split |
Episodes |
路径 |
|---|---|---|
|
778 |
|
|
1,839 |
|
完整目录如下:
streamvln_datasets/
├── datasets/r2r/
│ ├── val_seen/val_seen.json.gz
│ └── val_unseen/val_unseen.json.gz
└── scene_datasets/mp3d/
└── <scene-id>/
├── <scene-id>.glb
└── <scene-id>.navmesh
3.3 配置路径
export SWIFTVLN_HABITAT_DATA_ROOT="/path/to/streamvln_datasets"
export SWIFTVLN_HABITAT_SCENES_DIR="${SWIFTVLN_HABITAT_DATA_ROOT}/scene_datasets"
export SWIFTVLN_HABITAT_R2R_EVAL_DATA_PATH="${SWIFTVLN_HABITAT_DATA_ROOT}/datasets/r2r/{split}/{split}.json.gz"
3.4 校验数据
swiftvln validate-evaluation-data --env-type habitat
命令检查 Episode 数量、字段、重复 ID,以及 MP3D .glb 与 .navmesh 场景文件。