HitLyn/TAMS-Visual-Pushing

RL training code for our visual-pushing paper

Visual-Pushing

Visual Pushing Task on TAMS UR5 Robot

Install

• Prerequest:

  • mujoco
  • gym
  • robogym

• Install this package:

  git clone PATH_TO_THIS_PACKAGE.git
  cd Visual-Pushing
  export export VISUAL_PUSHING_HOME={CURRENT ABS PATH TO THIS PACKAGE}
  python setup.py develop

• Install Pytorch: https://pytorch.org/

Image Segmentation

Cycle GAN

Reinforcement Learning with HER