/Steering-Network-1

Primary LanguagePython

Steering-Network-1

train_sample

Create dataset

mkdir raw_dataset
  • Download the UBP dataset into the "raw_dataset" directory. A sample of the UPB dataset is available here.
mkdir scene_splits
  • Download the scene splits into the "scene_splits" directory. The train-validation split is available here. In the "scene_splits" directory you should have: "train_scenes.txt" and "test_scenes.txt".
cd scripts

# create the dataset
python3 create_dataset.py --root_dir ../raw_dataset

# split the dataset into train-test
python3 split_dataset.py --train ../scene_splits/train_scenes.txt --test ../scene_splits/test_scenes.txt

# create synthetic dataset by performin 2D perspective augmentations only for the training dataset
python3 create_aug_dataset.py --root_dir ../raw_dataset --train ../scene_splits/train_scenes.txt

# compute balancing weights
python3 weights.py
python3 weights.py --augm

Train models

./run_train.sh

Test models - Open-loop evaluation

./run_test.sh