add requirements

keineahnung2345 · Apr 6, 2018 · 8c47ad6 · 8c47ad6
1 parent 451d738
commit 8c47ad6
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 7 deletions.
diff --git a/README.md b/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:

diff --git a/requirements.txt b/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]
diff --git a/samples/balloon/README.md b/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
 ```