18 |
34 |
19 |
33 |
20 |
32 |
21 | cloud_upload
22 | Drag and drop image here
23 | or
24 |
30 |
31 |
35 |
45 |
36 | ![]()
37 |
38 |
39 | Upload and Detect Image
40 |
41 |
42 | Go Back
43 |
44 |
46 |
52 |
47 |
48 |
49 | Go Back
50 |
51 |
53 |
74 |
75 |
76 |
77 |
78 |
--------------------------------------------------------------------------------
/classify_image/views.py:
--------------------------------------------------------------------------------
1 | import io
2 | import os
3 |
4 | from base64 import b64decode
5 | import tensorflow as tf
6 | from PIL import Image
7 | from django.core.files.temp import NamedTemporaryFile
8 | from django.http import JsonResponse
9 | from django.views.decorators.csrf import csrf_exempt
10 | from django.shortcuts import render
11 |
12 | MAX_K = 10
13 |
14 | TF_GRAPH = "{base_path}/inception_model/graph.pb".format(
15 | base_path=os.path.abspath(os.path.dirname(__file__)))
16 | TF_LABELS = "{base_path}/inception_model/labels.txt".format(
17 | base_path=os.path.abspath(os.path.dirname(__file__)))
18 |
19 |
20 | def load_graph():
21 | sess = tf.Session()
22 | with tf.gfile.FastGFile(TF_GRAPH, 'rb') as tf_graph:
23 | graph_def = tf.GraphDef()
24 | graph_def.ParseFromString(tf_graph.read())
25 | tf.import_graph_def(graph_def, name='')
26 | label_lines = [line.rstrip() for line in tf.gfile.GFile(TF_LABELS)]
27 | softmax_tensor = sess.graph.get_tensor_by_name('softmax:0')
28 | return sess, softmax_tensor, label_lines
29 |
30 |
31 | SESS, GRAPH_TENSOR, LABELS = load_graph()
32 |
33 |
34 | @csrf_exempt
35 | def classify_api(request):
36 | data = {"success": False}
37 |
38 | if request.method == "POST":
39 | tmp_f = NamedTemporaryFile()
40 |
41 | if request.FILES.get("image", None) is not None:
42 | image_request = request.FILES["image"]
43 | image_bytes = image_request.read()
44 | image = Image.open(io.BytesIO(image_bytes))
45 | image.save(tmp_f, image.format)
46 | elif request.POST.get("image64", None) is not None:
47 | base64_data = request.POST.get("image64", None).split(',', 1)[1]
48 | plain_data = b64decode(base64_data)
49 | tmp_f.write(plain_data)
50 |
51 | classify_result = tf_classify(tmp_f, int(request.POST.get('k', MAX_K)))
52 | tmp_f.close()
53 |
54 | if classify_result:
55 | data["success"] = True
56 | data["confidence"] = {}
57 | for res in classify_result:
58 | data["confidence"][res[0]] = float(res[1])
59 |
60 | return JsonResponse(data)
61 |
62 |
63 | def classify(request):
64 | return render(request, 'classify.html', {})
65 |
66 |
67 | # noinspection PyUnresolvedReferences
68 | def tf_classify(image_file, k=MAX_K):
69 | result = list()
70 |
71 | image_data = tf.gfile.FastGFile(image_file.name, 'rb').read()
72 |
73 | predictions = SESS.run(GRAPH_TENSOR, {'DecodeJpeg/contents:0': image_data})
74 | predictions = predictions[0][:len(LABELS)]
75 | top_k = predictions.argsort()[-k:][::-1]
76 | for node_id in top_k:
77 | label_string = LABELS[node_id]
78 | score = predictions[node_id]
79 | result.append([label_string, score])
80 |
81 | return result
82 |
--------------------------------------------------------------------------------
/manage.py:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env python
2 | import os
3 | import sys
4 |
5 | if __name__ == "__main__":
6 | os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tf_inception.settings")
7 | try:
8 | from django.core.management import execute_from_command_line
9 | except ImportError:
10 | # The above import may fail for some other reason. Ensure that the
11 | # issue is really that Django is missing to avoid masking other
12 | # exceptions on Python 2.
13 | try:
14 | import django
15 | except ImportError:
16 | raise ImportError(
17 | "Couldn't import Django. Are you sure it's installed and "
18 | "available on your PYTHONPATH environment variable? Did you "
19 | "forget to activate a virtual environment?"
20 | )
21 | raise
22 | execute_from_command_line(sys.argv)
23 |
--------------------------------------------------------------------------------
/requirements.txt:
--------------------------------------------------------------------------------
1 | Django
2 | Pillow
3 | gunicorn
4 | tensorflow
5 | WhiteNoise
6 |
--------------------------------------------------------------------------------
/runtime.txt:
--------------------------------------------------------------------------------
1 | python-3.6.2
2 |
--------------------------------------------------------------------------------
/tf_inception/__init__.py:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/ccd97/image-classify-server/93a4bc9edec0c357f5b342721db9f9a456182419/tf_inception/__init__.py
--------------------------------------------------------------------------------
/tf_inception/settings.py:
--------------------------------------------------------------------------------
1 | """
2 | Django settings for TensorflowServer project.
3 |
4 | Generated by 'django-admin startproject' using Django 1.10.3.
5 |
6 | For more information on this file, see
7 | https://docs.djangoproject.com/en/1.10/topics/settings/
8 |
9 | For the full list of settings and their values, see
10 | https://docs.djangoproject.com/en/1.10/ref/settings/
11 | """
12 |
13 | import os
14 |
15 | # Build paths inside the project like this: os.path.join(BASE_DIR, ...)
16 | BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
17 |
18 | # Quick-start development settings - unsuitable for production
19 | # See https://docs.djangoproject.com/en/1.10/howto/deployment/checklist/
20 |
21 | # SECURITY WARNING: keep the secret key used in production secret!
22 | SECRET_KEY = '6t918%**s8@^(a5kl=y8ir@^goesxbp9#*c46=#hh#dldf8^fu'
23 |
24 | # SECURITY WARNING: don't run with debug turned on in production!
25 | DEBUG = False
26 |
27 | ALLOWED_HOSTS = [
28 | 'tf-classify.herokuapp.com',
29 | '127.0.0.1',
30 | ]
31 |
32 | # Application definition
33 |
34 | INSTALLED_APPS = [
35 | 'django.contrib.auth',
36 | 'django.contrib.contenttypes',
37 | 'django.contrib.sessions',
38 | 'django.contrib.messages',
39 | 'django.contrib.staticfiles',
40 | 'classify_image'
41 | ]
42 |
43 | MIDDLEWARE = [
44 | 'django.middleware.security.SecurityMiddleware',
45 | 'django.contrib.sessions.middleware.SessionMiddleware',
46 | 'django.middleware.common.CommonMiddleware',
47 | 'django.middleware.csrf.CsrfViewMiddleware',
48 | 'django.contrib.auth.middleware.AuthenticationMiddleware',
49 | 'django.contrib.messages.middleware.MessageMiddleware',
50 | 'django.middleware.clickjacking.XFrameOptionsMiddleware',
51 | 'whitenoise.middleware.WhiteNoiseMiddleware',
52 | ]
53 |
54 | ROOT_URLCONF = 'tf_inception.urls'
55 |
56 | TEMPLATES = [
57 | {
58 | 'BACKEND': 'django.template.backends.django.DjangoTemplates',
59 | 'DIRS': [],
60 | 'APP_DIRS': True,
61 | 'OPTIONS': {
62 | 'context_processors': [
63 | 'django.template.context_processors.debug',
64 | 'django.template.context_processors.request',
65 | 'django.contrib.auth.context_processors.auth',
66 | 'django.contrib.messages.context_processors.messages',
67 | ],
68 | },
69 | },
70 | ]
71 |
72 | # Database
73 | # https://docs.djangoproject.com/en/1.10/ref/settings/#databases
74 |
75 | DATABASES = {
76 | 'default': {
77 | 'ENGINE': 'django.db.backends.sqlite3',
78 | 'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
79 | }
80 | }
81 |
82 | # Password validation
83 | # https://docs.djangoproject.com/en/1.10/ref/settings/#auth-password-validators
84 |
85 | AUTH_PASSWORD_VALIDATORS = [
86 | {
87 | 'NAME': 'django.contrib.auth.password_validation.UserAttributeSimilarityValidator',
88 | },
89 | {
90 | 'NAME': 'django.contrib.auth.password_validation.MinimumLengthValidator',
91 | },
92 | {
93 | 'NAME': 'django.contrib.auth.password_validation.CommonPasswordValidator',
94 | },
95 | {
96 | 'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator',
97 | },
98 | ]
99 |
100 | # Internationalization
101 | # https://docs.djangoproject.com/en/1.10/topics/i18n/
102 |
103 | LANGUAGE_CODE = 'en-us'
104 |
105 | TIME_ZONE = 'UTC'
106 |
107 | USE_I18N = True
108 |
109 | USE_L10N = True
110 |
111 | USE_TZ = True
112 |
113 | PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
114 |
115 | # Static files (CSS, JavaScript, Images)
116 | # https://docs.djangoproject.com/en/1.9/howto/static-files/
117 | STATIC_ROOT = os.path.join(PROJECT_ROOT, 'staticfiles')
118 | STATIC_URL = '/static/'
119 |
120 | # Extra places for collectstatic to find static files.
121 | STATICFILES_DIRS = (
122 | os.path.join(BASE_DIR, 'classify_image/static'),
123 | )
124 |
125 | STATICFILES_STORAGE = 'whitenoise.storage.CompressedManifestStaticFilesStorage'
126 |
--------------------------------------------------------------------------------
/tf_inception/urls.py:
--------------------------------------------------------------------------------
1 | """TensorflowServer URL Configuration
2 |
3 | The `urlpatterns` list routes URLs to views. For more information please see:
4 | https://docs.djangoproject.com/en/1.10/topics/http/urls/
5 | Examples:
6 | Function views
7 | 1. Add an import: from my_app import views
8 | 2. Add a URL to urlpatterns: url(r'^$', views.home, name='home')
9 | Class-based views
10 | 1. Add an import: from other_app.views import Home
11 | 2. Add a URL to urlpatterns: url(r'^$', Home.as_view(), name='home')
12 | Including another URLconf
13 | 1. Import the include() function: from django.conf.urls import url, include
14 | 2. Add a URL to urlpatterns: url(r'^blog/', include('blog.urls'))
15 | """
16 | from django.conf.urls import url
17 | from django.views.generic import RedirectView
18 | # from django.contrib import admin
19 | from classify_image.views import classify, classify_api
20 |
21 | urlpatterns = [
22 | url(r'^classify_image/classify/api/$', classify_api),
23 | url(r'^classify_image/classify/$', classify),
24 | url(r'^$', RedirectView.as_view(url='classify_image/classify/'))
25 | # url(r'^admin/', admin.site.urls),
26 | ]
27 |
--------------------------------------------------------------------------------