Skip to content

Latest commit

 

History

History
86 lines (52 loc) · 3.8 KB

README.md

File metadata and controls

86 lines (52 loc) · 3.8 KB
Raw

tesse-gym

Provides a Python interface for reinforcement learning using the TESSE Unity environment and the OpenAI Gym toolkit.

GOSEEK Navigation

Installation

From Source

Using Anaconda or miniconda is highly recommended. Python 3.7 is required.

  1. If using conda, create a new environment
conda create -n goseek python=3.7 ipython jupyter numpy scipy
conda activate goseek
  1. Clone and install this repository.
git clone https://github.com/MIT-TESSE/tesse-gym.git
cd tesse-gym

python setup.py install

cd ..

GOSEEK Challenge

The objective of this task is to navigate an agent through an office environment to collect randomly-spawned fruit as quickly as possible.

GOSEEK Teaser Trailer

More specifically, the agent can select from one of four actions at each decision epoch: move forward 0.5 meters, turn left 8 degrees, turn right 8 degrees, and collect fruit within 2.0 meters of the agent's current position. Our robot is equiped with stereo cameras and an Inertial Measurement Unit (IMU), from which a state-of-the-art perception pipeline estimates three pieces of information that make up the agent's observation at each decision epoch: localization information (position and heading relative to start position), pixel-wise semantic labels for objects in the robot's field of view, and pixel-wise depth in the robot's field of view.

Quick Start

To run the example PPO baseline, install Stable Baselines and a version of Tensorflow between v0.8.0 and v1.14.0 (see the Stable Baselines docs for details).

Note: Currently, we've tested Python 3.7 Tensorflow installation with Anaconda against Cuda 10.0-10.2 (run nvcc -V to check your Cuda version).

For Cuda 10.0, we'd recommend installing tensorflow-gpu v1.13.1:

conda activate goseek 
conda install tensorflow-gpu==1.13.1

For Cuda 10.1 and 10.2, we'd recommend installing tensorflow-gpu v1.14:

conda activate goseek 
conda install tensorflow-gpu==1.14

Then, install Stable Baselines

conda activate goseek 
pip install stable-baselines

Then, see the example notebook to train an agent.

Challenge Details

See the GOSEEK Challenge landing page for details on setup, evaluation, and submission.

Disclaimer

DISTRIBUTION STATEMENT A. Approved for public release. Distribution is unlimited.

This material is based upon work supported by the Under Secretary of Defense for Research and Engineering under Air Force Contract No. FA8702-15-D-0001. Any opinions, findings, conclusions or recommendations expressed in this material are those of the author(s) and do not necessarily reflect the views of the Under Secretary of Defense for Research and Engineering.

© 2020 Massachusetts Institute of Technology.

MIT Proprietary, Subject to FAR52.227-11 Patent Rights - Ownership by the contractor (May 2014)

The software/firmware is provided to you on an As-Is basis

Delivered to the U.S. Government with Unlimited Rights, as defined in DFARS Part 252.227-7013 or 7014 (Feb 2014). Notwithstanding any copyright notice, U.S. Government rights in this work are defined by DFARS 252.227-7013 or DFARS 252.227-7014 as detailed above. Use of this work other than as specifically authorized by the U.S. Government may violate any copyrights that exist in this work.