Merge pull request #16 from vicrep/master

Add pip requirements
tonybeltramelli · Dec 13, 2017 · 94f651f · 94f651f
2 parents 64bc3b3 + 2045ed6
commit 94f651f
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -152,3 +152,10 @@ com_crashlytics_export_strings.xml
 crashlytics.properties
 crashlytics-build.properties
 fabric.properties
+
+# Generated dataset files
+datasets/*
+!datasets/pix2code_datasets.z*
+
+# Training output
+bin/
diff --git a/README.md b/README.md
@@ -30,6 +30,18 @@ The current implementation is not, in any way, intended, nor able to generate co
 We could not emphasize enough that this project is experimental and shared for educational purposes only.
 Both the source code and the datasets are provided to foster future research in machine intelligence and are not designed for end users.
 
+## Setup
+### Prerequisites
+
+- Python 2 or 3
+- pip
+
+### Install dependencies
+
+```sh
+pip install -r  requirements.txt
+```
+
 ## Usage
 
 Prepare the data:

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,5 @@
+Keras==2.1.2
+numpy==1.13.3
+opencv-python==3.3.0.10
+h5py==2.7.1
+tensorflow==1.4.0