add requirements

README.md

@@ -161,10 +161,10 @@ You can also [join our team](https://matterport.com/careers/) and help us build
 
 ## Requirements
 * Python 3.4+
-* TensorFlow 1.3+
-* Keras 2.0.8+
-* Jupyter Notebook
-* Numpy, skimage, scipy, Pillow, cython, h5py
+
+```bash
+pip3 install -r requirements.txt
+```
 
 ### MS COCO Requirements:
 To train or test on MS COCO, you'll also need:

requirements.txt

@@ -0,0 +1,12 @@
+numpy
+scipy
+Pillow
+cython
+matplotlib
+scikit-image
+tensorflow>=1.3.0
+keras>=2.0.8
+opencv-python
+h5py
+imgaug
+IPython[all]

samples/balloon/README.md

@@ -30,17 +30,17 @@ Open the `inspect_balloon_data.ipynb` or `inspect_balloon_model.ipynb` Jupter no
 ## Train the Balloon model
 
 Train a new model starting from pre-trained COCO weights
-```
+```bash
 python3 balloon.py train --dataset=/path/to/balloon/dataset --weights=coco
 ```
 
 Resume training a model that you had trained earlier
-```
+```bash
 python3 balloon.py train --dataset=/path/to/balloon/dataset --weights=last
 ```
 
 Train a new model starting from ImageNet weights
-```
+```bash
 python3 balloon.py train --dataset=/path/to/balloon/dataset --weights=imagenet
 ```