correct prediction

Author: chen0040

.gitignore

@@ -32,6 +32,8 @@ wheels/
 
 Thumbs.db
 
+keras_image_classifier_web/uploads
+
 # PyInstaller
 #  Usually these files are written by a python script from a template
 #  before PyInstaller builds the exe, so as to inject date/other infos into it.

keras_image_classifier/cnn_bi_classifier_predict.py

@@ -1,8 +1,32 @@
 from keras.models import model_from_json
+from keras.preprocessing.image import load_img, img_to_array
+from PIL import Image
+import numpy as np
 
-model = model_from_json('models/cnn_bi_classifier_architecture.json')
+json = open('models/cnn_bi_classifier_architecture.json', 'r').read()
+
+model = model_from_json(json)
 model.load_weights('models/cnn_bi_classifier_weights.h5')
 
 model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics=['accuracy'])
 
-model.predict_classes()
+def predict(filename, label):
+    img = Image.open(filename)
+    img = img.resize((150, 150), Image.ANTIALIAS)
+
+    input = np.asarray(img)
+    input = np.expand_dims(input, axis=0)
+
+    print(input.shape)
+
+    output = model.predict(input)
+
+    print('This is a ' + label)
+    print('Probability that it is a dog ' + str(output[0][0]))
+
+
+for i in range(100):
+    predict('bi_classifier_data/training/cat/cat.' + str(i) + '.jpg', 'cat')
+    predict('bi_classifier_data/training/dog/dog.' + str(i) + '.jpg', 'dog')
+
+

keras_image_classifier_web/flaskr.py

@@ -1,115 +1,46 @@
 import os
+from keras.models import model_from_json
 import sqlite3
 import uuid
 
 from flask import Flask, request, session, g, redirect, url_for, abort, \
-     render_template, flash
+    render_template, flash
 from flask import send_from_directory
 from werkzeug.utils import secure_filename
 
 UPLOAD_FOLDER = '../uploads'
 ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
 
-app = Flask(__name__) # create the application instance :)
-app.config.from_object(__name__) # load config from this file , flaskr.py
+app = Flask(__name__)  # create the application instance :)
+app.config.from_object(__name__)  # load config from this file , flaskr.py
 
 # Load default config and override config from an environment variable
-app.config.update(dict(
-    DATABASE=os.path.join(app.root_path, 'keras_image_projects.db'),
-    SECRET_KEY='development-key',
-    USERNAME='root',
-    PASSWORD='chen0469'
-))
 app.config.from_envvar('FLASKR_SETTINGS', silent=True)
 app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
 
+model = model_from_json(open(os.path.join('models', '/cnn_bi_classifier_architecture.json')).read())
+model.load_weights(os.path.join('models', '/cnn_bi_classifier_weights.h5'))
 
-def connect_db():
-    """Connects to the specific database."""
-    rv = sqlite3.connect(app.config['DATABASE'])
-    rv.row_factory = sqlite3.Row
-    return rv
-
-
-def get_db():
-    """Opens a new database connection if there is none yet for the
-    current application context.
-    """
-    if not hasattr(g, 'sqlite_db'):
-        g.sqlite_db = connect_db()
-    return g.sqlite_db
-
-
-@app.teardown_appcontext
-def close_db(error):
-    """Closes the database again at the end of the request."""
-    if hasattr(g, 'sqlite_db'):
-        g.sqlite_db.close()
-
-
-def init_db():
-    db = get_db()
-    with app.open_resource('schema.sql', mode='r') as f:
-        db.cursor().executescript(f.read())
-    db.commit()
-
-
-@app.cli.command('initdb')
-def initdb_command():
-    """Initializes the database."""
-    init_db()
-    print('Initialized the database.')
+model.compile(optimizer='rmsprop', loss='binary_crossentropy', metrics=['accuracy'])
 
 
 @app.route('/')
-def show_entries():
-    db = get_db()
-    cur = db.execute('select id, description, name from projects order by id desc')
-    entries = cur.fetchall()
-    return render_template('show_entries.html', entries=entries)
-
-
-@app.route('/add', methods=['POST'])
-def add_entry():
-    if not session.get('logged_in'):
-        abort(401)
-    db = get_db()
-    db.execute('insert into projects (name, description) values (?, ?)',
-                 [request.form['name'], request.form['description'], str(uuid.uuid4())])
-    db.commit()
-    flash('New entry was successfully posted')
-    return redirect(url_for('show_entries'))
-
-
-@app.route('/login', methods=['GET', 'POST'])
-def login():
-    error = None
-    if request.method == 'POST':
-        if request.form['username'] != app.config['USERNAME']:
-            error = 'Invalid username'
-        elif request.form['password'] != app.config['PASSWORD']:
-            error = 'Invalid password'
-        else:
-            session['logged_in'] = True
-            flash('You were logged in')
-            return redirect(url_for('show_entries'))
-    return render_template('login.html', error=error)
-
-
-@app.route('/logout')
-def logout():
-    session.pop('logged_in', None)
-    flash('You were logged out')
-    return redirect(url_for('show_entries'))
+def classifiers():
+    return render_template('classifiers.html')
 
 
 def allowed_file(filename):
     return '.' in filename and \
            filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
 
-@app.route('/projects/<token>/images', methods=['GET', 'POST'])
-def upload_file(token):
+@app.route('/about', methods=['GET'])
+def about():
+    return 'about us'
+
+
+@app.route('/cats_vs_dogs', methods=['GET', 'POST'])
+def cats_vs_dogs():
     if request.method == 'POST':
         # check if the post request has the file part
         if 'file' not in request.files:
@@ -124,29 +55,22 @@ def upload_file(token):
         if file and allowed_file(file.filename):
             filename = secure_filename(file.filename)
             file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
-            return redirect(url_for('uploaded_file',
+            return redirect(url_for('cats_vs_dogs_result',
                                     filename=filename))
-    return '''
-    <!doctype html>
-    <title>Upload new File</title>
-    <h1>Upload new File</h1>
-    <form method=post enctype=multipart/form-bi_classifier_data>
-      <p><input type=file name=file>
-         <input type=submit value=Upload>
-    </form>
-    '''
-
-
-@app.route('/uploads/<filename>')
-def uploaded_file(filename):
+    return render_template('cats_vs_dogs.html')
+
+
+@app.route('/cats_vs_dogs_result/<filename>')
+def cats_vs_dogs_result(filename):
+    filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+    return render_template('cats_vs_dogs_result.html', filename=filename)
+
+
+@app.route('/images/<filename>')
+def get_image(filename):
     return send_from_directory(app.config['UPLOAD_FOLDER'],
                                filename)
 
-@app.before_first_request
-def startup():
-    init_db()
-
 
 if __name__ == '__main__':
     app.run(debug=True)
-

keras_image_classifier_web/models/cnn_bi_classifier_architecture.json

@@ -0,0 +1 @@
+{"class_name": "Sequential", "config": [{"class_name": "Conv2D", "config": {"name": "conv2d_1", "trainable": true, "batch_input_shape": [null, 150, 150, 3], "dtype": "float32", "filters": 32, "kernel_size": [3, 3], "strides": [1, 1], "padding": "valid", "data_format": "channels_last", "dilation_rate": [1, 1], "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "VarianceScaling", "config": {"scale": 1.0, "mode": "fan_avg", "distribution": "uniform", "seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Activation", "config": {"name": "activation_1", "trainable": true, "activation": "relu"}}, {"class_name": "MaxPooling2D", "config": {"name": "max_pooling2d_1", "trainable": true, "pool_size": [2, 2], "padding": "valid", "strides": [2, 2], "data_format": "channels_last"}}, {"class_name": "Conv2D", "config": {"name": "conv2d_2", "trainable": true, "filters": 32, "kernel_size": [3, 3], "strides": [1, 1], "padding": "valid", "data_format": "channels_last", "dilation_rate": [1, 1], "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "VarianceScaling", "config": {"scale": 1.0, "mode": "fan_avg", "distribution": "uniform", "seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Activation", "config": {"name": "activation_2", "trainable": true, "activation": "relu"}}, {"class_name": "MaxPooling2D", "config": {"name": "max_pooling2d_2", "trainable": true, "pool_size": [2, 2], "padding": "valid", "strides": [2, 2], "data_format": "channels_last"}}, {"class_name": "Conv2D", "config": {"name": "conv2d_3", "trainable": true, "filters": 64, "kernel_size": [3, 3], "strides": [1, 1], "padding": "valid", "data_format": "channels_last", "dilation_rate": [1, 1], "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "VarianceScaling", "config": {"scale": 1.0, "mode": "fan_avg", "distribution": "uniform", "seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Activation", "config": {"name": "activation_3", "trainable": true, "activation": "relu"}}, {"class_name": "MaxPooling2D", "config": {"name": "max_pooling2d_3", "trainable": true, "pool_size": [2, 2], "padding": "valid", "strides": [2, 2], "data_format": "channels_last"}}, {"class_name": "Flatten", "config": {"name": "flatten_1", "trainable": true}}, {"class_name": "Dense", "config": {"name": "dense_1", "trainable": true, "units": 64, "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "VarianceScaling", "config": {"scale": 1.0, "mode": "fan_avg", "distribution": "uniform", "seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Dropout", "config": {"name": "dropout_1", "trainable": true, "rate": 0.5}}, {"class_name": "Dense", "config": {"name": "dense_2", "trainable": true, "units": 1, "activation": "linear", "use_bias": true, "kernel_initializer": {"class_name": "VarianceScaling", "config": {"scale": 1.0, "mode": "fan_avg", "distribution": "uniform", "seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Activation", "config": {"name": "activation_4", "trainable": true, "activation": "sigmoid"}}], "keras_version": "2.0.5", "backend": "tensorflow"}

keras_image_classifier_web/models/cnn_bi_classifier_weights.h5

@@ -0,0 +1,7 @@
+{% extends "layout.html" %}
+{% block body %}
+    <form method=post enctype=multipart/form-data>
+      <p><input type=file name=file>
+         <input type=submit value=Upload>
+    </form>
+{% endblock %}

keras_image_classifier_web/templates/cats_vs_dogs.html

@@ -0,0 +1,7 @@
+{% extends "layout.html" %}
+{% block body %}
+    <form method=post enctype=multipart/form-data>
+      <p><input type=file name=file>
+         <input type=submit value=Upload>
+    </form>
+{% endblock %}

keras_image_classifier_web/templates/cats_vs_dogs_result.html

@@ -0,0 +1,6 @@
+{% extends "layout.html" %}
+{% block body %}
+  <ul class=entries>
+  <li><a href="{{ url_for('cats_vs_dogs') }}">Cats vs Dogs</a></li>
+  </ul>
+{% endblock %}

keras_image_classifier_web/templates/classifiers.html

@@ -1,14 +1,10 @@
 <!doctype html>
-<title>Keras Image Classifier Web</title>
+<title>Flask Slingshot</title>
 <link rel=stylesheet type=text/css href="{{ url_for('static', filename='style.css') }}">
 <div class=page>
-  <h1>Keras Image Classifier</h1>
+  <h1><a href="/">Image Classifier</a></h1>
   <div class=metanav>
-  {% if not session.logged_in %}
-    <a href="{{ url_for('login') }}">log in</a>
-  {% else %}
-    <a href="{{ url_for('logout') }}">log out</a>
-  {% endif %}
+  <a href="{{ url_for('about') }}">About</a>
   </div>
   {% for message in get_flashed_messages() %}
     <div class=flash>{{ message }}</div>