StanfordVL / behavioral_navigation_nlp

Code for translating navigation instructions in natural language to a high-level plan for behavioral navigation for robot navigation
9 stars 2 forks source link

Translating Navigation Instructions in Natural Language to a High-Level Plan for Behavioral Robot Navigation

Xiaoxue Zang, Ashwini Pokle, Marynel Vázquez, Kevin Chen, Juan Carlos Niebles, Alvaro Soto, Silvio Savarese

* equal contribution.

Setup

Run the following command to install dependencies and libraries, download glove embeddings and dataset for experiments.

sh get_started.sh

Usage

Training

python codes/main.py --experiment_name=[unique name for model folder] --data_dir data

You can also download the Trained model

Test

Evaluate the F1 score and Exact Match on the dev/test dataset, which can be decided by --file_in_path.

python codes/main.py --train_dir=experiments/[experiment name] --mode=official_eval --data_dir data --file_in_path [dev/test]

--file_out_path is needed if you want to write the prediction out to a file. [--write_out=True]

Run the following instruction to evaluate the result and plot the error analysis.

python codes/evaluate.py ground_truth_file prediction_file

Show examples

Show the predictions of some examples. --file_in_path decides the dataset to test with. There are three valid arguments: dev, test, test_diff_maps, which respectively mean validation set, test-repeated map set, test-new map set.

python codes/main.py --train_dir=experiments/[experiment name] --mode=show_examples --data_dir data --file_in_path [dev/test/test_diff_maps] --print_num 10

Examples

Example training commmand:

python codes/main.py --experiment_name=my_experiment --mode=train --data_dir my_data

Example test command:

python codes/main.py --train_dir experiments/my_experiment --mode=official_eval --data_dir my_data

Acknowledgement

Toyota Research Institute ("TRI") provided funds to assist the authors with their research but this article solely reflects the opinions and conclusions of its authors and not TRI or any other Toyota entity.

code modified from the Default Final Project for CS224n, Winter 2018