Skip to content

Latest commit

 

History

History
67 lines (41 loc) · 1.89 KB

README.md

File metadata and controls

67 lines (41 loc) · 1.89 KB

Visual Imitation Made Easy

Sarah Young1, Dhiraj Gandhi2, Shubham Tulsiani2, Abhinav Gupta2 3, Pieter Abbeel1, Lerrel Pinto1 4

1University of California, Berkeley, 2Facebook AI Research, 3Carnegie Mellon University, 4New York University

Data

Pushing data is located here.

Stacking data is located here.

Usage

Below are example scripts for training and testing on provided sample data.

Setup

  1. Clone repo.
git clone https://github.com/sarahisyoung/Visual-Imitation-Made-Easy.git
  1. Create and activate conda environment.
conda env create -f environment.yml
conda activate trashbot
  1. Set env path.
export PYTHONPATH=$PYTHONPATH:path_to_proj/

Training

To train with custom data, see this for details on data processing.

  1. Download the data (~45 GB) from dropbox by running this script. Feel free to comment out training/validation URL in the script to download just a small sample of the data.

    python download_data.py --task push
  2. To train:

    python train.py --task push --train_dir data/train --val_dir data/val --test_dir data/test --save_dir results

Testing

  1. This command predicts on a folder of images. Output visualization is saved to specified folder.
python push_test.py --model results/exp1/policy_earlystop.pt --image_folder test_data/ --output predicted_data/