initial commit

Author: Intesar Haider

.gitignore

@@ -0,0 +1,44 @@
+# Compiled source #
+###################
+*.com
+*.class
+*.dll
+*.exe
+*.o
+*.so
+*.pyc
+
+# Packages #
+############
+# it's better to unpack these files and commit the raw source
+# git has its own built in compression methods
+*.7z
+*.dmg
+*.gz
+*.iso
+*.rar
+#*.tar
+*.zip
+
+# Logs and databases #
+######################
+*.log
+*.sqlite
+
+# OS generated files #
+######################
+.DS_Store
+ehthumbs.db
+Icon
+Thumbs.db
+.tmtags
+.idea
+tags
+vendor.tags
+tmtagsHistory
+*.sublime-project
+*.sublime-workspace
+.bundle
+
+env
+models

ModelType.py

@@ -0,0 +1,5 @@
+from enum import Enum
+
+class ModelType(Enum):
+    car_make = "models/car_make/"
+    car_model = "models/car_model/"

Procfile

@@ -0,0 +1 @@
+web: gunicorn app:app

README.md

@@ -0,0 +1,4 @@
+Upload file to predict car from images
+======================================
+
+Image classification

app.py

@@ -0,0 +1,58 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import os
+from datetime import datetime
+from flask import Flask, render_template, jsonify, request
+
+import trained_model as tm
+import ModelType
+app = Flask(__name__)
+app.config.from_object(__name__)
+app.config['UPLOAD_FOLDER'] = 'uploads'
+
+ALLOWED_EXTENSIONS = ['jpg', 'jpeg']
+
+carMakeModel, carMakeGraph, carMakeLabel = tm.load_model(ModelType.ModelType.car_make)
+carModelModel, carModelGraph, carModelLabel = tm.load_model(ModelType.ModelType.car_model)
+
+@app.route("/")
+def index():
+    return render_template("index.html")
+
+
+@app.route("/car_make")
+def car_make():
+    return render_template("car_make.html")
+
+
+@app.route("/car_model")
+def car_model():
+    return render_template("car_model.html")
+
+@app.route('/upload/<string:model_type>', methods=['POST'])
+def upload(model_type):
+    if request.method == 'POST':
+        file = request.files['file']
+        if file and allowed_file(file.filename):
+            now = datetime.now()
+            filename = os.path.join(app.config['UPLOAD_FOLDER'], "%s.%s" % (now.strftime("%Y-%m-%d-%H-%M-%S-%f"), file.filename.rsplit('.', 1)[1]))
+            file.save(filename)
+
+            if ModelType.ModelType.car_model.name == model_type:
+                prediction = tm.predict(carModelGraph, carModelModel, filename, carModelLabel)
+            elif ModelType.ModelType.car_make.name == model_type:
+                prediction = tm.predict(carMakeGraph, carMakeModel, filename, carMakeLabel)
+            else:
+                os.remove(filename)
+                return jsonify("wrong model type")
+
+            os.remove(filename)
+
+            return jsonify(prediction)
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
+
+if __name__ == "__main__":
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,39 @@
+absl-py==0.7.1
+astor==0.8.0
+Click==7.0
+cycler==0.10.0
+decorator==4.4.0
+Flask==1.1.1
+gast==0.2.2
+google-pasta==0.1.7
+grpcio==1.22.0
+gunicorn==19.9.0
+h5py==2.9.0
+imageio==2.5.0
+itsdangerous==1.1.0
+Jinja2==2.10.1
+Keras==2.2.4
+Keras-Applications==1.0.8
+Keras-Preprocessing==1.1.0
+kiwisolver==1.1.0
+Markdown==3.1.1
+MarkupSafe==1.1.1
+matplotlib==3.1.1
+networkx==2.3
+numpy==1.16.4
+opencv-python==3.3.0.9
+Pillow==6.1.0
+protobuf==3.9.0
+pyparsing==2.4.0
+python-dateutil==2.8.0
+PyWavelets==1.0.3
+PyYAML==5.1.1
+scikit-image==0.15.0
+scipy==1.3.0
+six==1.12.0
+tensorboard==1.14.0
+tensorflow==1.14.0
+tensorflow-estimator==1.14.0
+termcolor==1.1.0
+Werkzeug==0.15.5
+wrapt==1.11.2

static/css/styles.css

@@ -0,0 +1,103 @@
+*{
+	margin: 0;
+	padding: 0;
+}
+
+html{
+	min-height: 100%;
+	position: relative;
+	
+}
+
+body{
+    color: grey;
+	background-color:#333;
+	min-height: 600px;
+	font: 14px/1.3 'Segoe UI',Arial, sans-serif;
+}
+
+#dropbox{
+    border: 1px solid grey;
+	border-radius: 3px;
+	position: relative;
+	margin: 80px auto 90px;
+	min-height: 290px;
+	overflow: hidden;
+	padding-bottom: 40px;
+    width: 990px;
+}
+
+
+#dropbox .message{
+	font-size: 11px;
+    text-align: center;
+    padding-top: 160px;
+    display: block;
+}
+
+#dropbox:before{
+	border-radius: 3px 3px 0 0;
+}
+
+#dropbox .preview{
+	width: 245px;
+	height: 215px;
+	float: left;
+	margin: 55px 0 0 60px;
+	position: relative;
+	text-align: center;
+}
+
+#dropbox .preview img{
+	max-width: 240px;
+	max-height: 180px;
+	border: 3px solid #fff;
+	display: block;
+	box-shadow: 0 0 2px #000;
+}
+
+#dropbox .imageHolder{
+	display: inline-block;
+	position: relative;
+}
+
+#dropbox .uploaded{
+	position: absolute;
+	top: 0;
+	left: 0;
+	height: 100%;
+	width: 100%;
+	background: url('../img/done.png') no-repeat center center rgba(255,255,255,0.5);
+	display: none;
+}
+
+#dropbox .preview.done .uploaded{
+	display: block;
+}
+
+#dropbox .progressHolder{
+	position: absolute;
+	background-color:#252f38;
+	height: 12px;
+	width: 100%;
+	left: 0;
+	bottom: 0;
+	box-shadow: 0 0 2px #000;
+}
+
+#dropbox .progress{
+	background-color:#2586d0;
+	position: absolute;
+	height: 100%;
+	left: 0;
+	width: 0;
+	box-shadow: 0 0 1px rgba(255, 255, 255, 0.4) inset;
+	-moz-transition: 0.25s;
+	-webkit-transition: 0.25s;
+	-o-transition: 0.25s;
+	transition: 0.25s;
+}
+
+#dropbox .preview.done .progress{
+	width: 100% !important;
+}