get vgg16 to work

Author: chen0040

README.md

@@ -1,3 +1,27 @@
 # keras-image-classifier-web-api
 
 This provides keras DCNN-based image classifiers codes with flask as the web api server for the image classifiers.
+
+# Usage
+
+Run the following command to install the keras, flask and other dependency modules:
+
+```bash
+sudo pip install -r requirements.txt
+```
+
+Goto keras_image_classifier_web directory and run the following command:
+
+```bash
+python flaskr.py
+```
+
+Now navigate your browser to http://localhost:5000 and you can try out various predictors built with the following
+trained classifiers:
+
+* binary classifier on "cats vs dogs" data
+* multi-class DCNN classifier trained with CIFAR-10 data
+* multi-class VGG16 classifier trained with ImageNet data
+
+
+

keras_image_classifier/cnn_vgg16_predict.py

@@ -0,0 +1,24 @@
+from keras.models import Model
+from keras.applications.vgg16 import VGG16, preprocess_input, decode_predictions
+from keras.optimizers import SGD
+from PIL import Image
+from keras.preprocessing.image import img_to_array
+import numpy as np
+
+model = VGG16(include_top=True, weights='imagenet')
+model.compile(optimizer=SGD(), loss='categorical_crossentropy', metrics=['accuracy'])
+
+
+def predict(filename):
+    img = Image.open(filename)
+    img = img.resize((224, 224), Image.ANTIALIAS)
+    input = img_to_array(img)
+    input = np.expand_dims(input, axis=0)
+    input = preprocess_input(input)
+    output = decode_predictions(model.predict(input), top=3)
+    print(output)
+
+
+for i in range(100):
+    predict('bi_classifier_data/training/cat/cat.' + str(i) + '.jpg')
+    predict('bi_classifier_data/training/dog/dog.' + str(i) + '.jpg')

keras_image_classifier_web/schema.sql

@@ -0,0 +1,7 @@
+Flask == 0.12.2
+gevent
+keras
+numpy
+h5py
+pillow
+https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

requirements.txt

@@ -6,6 +6,7 @@
     include_package_data=True,
     install_requires=[
         'flask',
+        'keras',
     ],
     setup_requires=[
         'pytest-runner',