GitXplorerGitXplorer
m

rock_detector

public
5 stars
0 forks
3 issues

Commits

List of commits on branch master.
Unverified
5442cd1d9adff858d6f01ce853d026ccafab4619

minor updates. the images are surreal these days

mmatwilso committed 7 years ago
Unverified
280f100781abd50a131cc2531754a998e76b18d7

light discs turned out great.

mmatwilso committed 7 years ago
Unverified
2351329adde7e1486c33478c002b138cab663d33

gumdrop button commit

mmatwilso committed 7 years ago
Unverified
aa9bc77c06a4ccfe89f795575d0fd8149d7328dd

rainbow splash commit

mmatwilso committed 7 years ago
Unverified
12d0f37ad3b4f298ccce7d3166a83289d0e2bedc

add docs folder

mmatwilso committed 7 years ago
Unverified
d54558755831c8071aa000708d3dde029f8f5e58

add data. update todos

mmatwilso committed 7 years ago

README

The README file for this repository.

rock_detector

[IN PROGRESS]
Domain randomization to train a convolutional neural net (VGG16) rock detector using Tensorflow and Mujoco, for NASA Robotic Mining Competition

Introduction

Check out this OpenAI blog post for an intro to Domain Randomization: https://blog.openai.com/spam-detection-in-the-physical-world/

I followed the details in this paper: https://arxiv.org/pdf/1703.06907.pdf

There was some nice stuff in OpenAI's mujoco_py repo to make this easier

Demo

Real robot view
test

YouTube video showing domain randomization

rock detector. exciting uh?

Installation

Install mujoco_py

READ THIS if you are getting ERROR: GLEW initialization error: Missing GL version

This was a huge headache and I had to do some hacky stuff to get it to work on my Ubuntu desktop. Mac was really easy to get it working and I would guess putting it in the cloud is not bad, because I'm pretty sure that's the way the folks at OpenAI are doing it.-

To get it to work, I cloned the repo, modified some source files, ran sudo python3 setup.py install in the folder and it somewhat worked.

git clone https://github.com/openai/mujoco-py.git
cd mujoco_py
pip3 install -r requirements.txt
sudo python3 setup.py install

Then, there will be many errors to fix, but most several are fixed by apt-get getting or pip installing something.

After that, I was still getting some errors, like this. I had to do some hacky stuff and I am not using GPU rendering, but it is running. If you have any issues, feel free to reach out to me.

Install blender

sudo apt-get install blender

Test it with

blender

If you are running on your local desktop (w/ monitor):

Use the file in assets/add_mesh_rock.zip and follow: https://blender.stackexchange.com/questions/8746/how-can-i-make-unique-rocks-in-blender-without-having-to-model-them-by-hand

Then activate it by clicking the check mark

[untested] Alternatively, if you are running headless (in cloud):

cp /path/to/rock_detector/assets/add_mesh_rocks.zip ~/.config/blender/<VERSION>/scripts/addons/

cd ~/.config/blender/<VERSION>/scripts/addons
unzip add_mesh_rocks.zip

Mujoco Tips

  <visual>
    <map znear=0.01 /> 
  </visual>