Skip to content

sarahisyoung/Visual-Imitation-Made-Easy

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Visual Imitation Made Easy

[Project Page] [Paper] [Video]

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/

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages