diff --git a/README.md b/README.md index e73b798b86..c8bfac6755 100644 --- 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 new file mode 100644 index 0000000000..bd64785cab --- /dev/null +++ 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] \ No newline at end of file diff --git a/samples/balloon/README.md b/samples/balloon/README.md index 67d9257626..4756d834d7 100644 --- 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 ```