MGL_VN_2024
收藏数据集概述
数据集内容
- thor_offline_data: 包含AI2-THOR场景中每个房间的所有可能位置的ResNet特征,以及场景的元数据和导航网络图。
- thor_glove: 包含导航目标的GloVe嵌入。
- gcn: 包含用于Scene Priors中图卷积网络(GCN)的数据,包括邻接矩阵。
- test_val_split: 包含测试和验证集的起始位置和场景信息。
数据集扩展
- 若需访问RGB图像,可替换
thor_offline_data为thor_offline_data_with_images,并通过命令行参数--images_file_name images.hdf5运行模型。
模型评估
使用预训练模型
-
SAVN: bash python main.py --eval --test_or_val test --episode_type TestValEpisode --load_model pretrained_models/savn_pretrained.dat --model SAVN --results_json savn_test.json cat savn_test.json
-
Scene Priors: bash python main.py --eval --test_or_val test --episode_type TestValEpisode --load_model pretrained_models/gcn_pretrained.dat --model GCN --glove_dir ./data/gcn --results_json scene_priors_test.json cat scene_priors_test.json
-
Non-Adaptive A3C: bash python main.py --eval --test_or_val test --episode_type TestValEpisode --load_model pretrained_models/nonadaptivea3c_pretrained.dat --results_json nonadaptivea3c_test.json cat nonadaptivea3c_test.json
训练模型
-
SAVN: bash python main.py --title savn_train --model SAVN --gpu-ids 0 1 --workers 12
-
Non-Adaptive A3C: bash python main.py --title nonadaptivea3c_train --gpu-ids 0 1 --workers 12
评估自训练模型
-
SAVN: bash python full_eval.py --title savn --model SAVN --results_json savn_results.json --gpu-ids 0 1 cat savn_results.json
-
Non-Adaptive A3C: bash python full_eval.py --title nonadaptivea3c --results_json nonadaptivea3c_results.json --gpu-ids 0 1 cat nonadaptivea3c_results.json
随机代理评估
- Random Agent: bash python main.py --eval --test_or_val test --episode_type TestValEpisode --title random_test --agent_type RandomNavigationAgent --results_json random_results.json cat random_results.json




