Skip to content

Commit 4fc56ae

Browse files
author
blueskymobile
committed
first commit
0 parents  commit 4fc56ae

24 files changed

+331
-0
lines changed

adrian.jpg

56.7 KB
Loading

cvApi/__init__.py

Whitespace-only changes.

cvApi/__init__.pyc

149 Bytes
Binary file not shown.

cvApi/settings.py

Lines changed: 120 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,120 @@
1+
"""
2+
Django settings for cvApi project.
3+
4+
Generated by 'django-admin startproject' using Django 1.10.4.
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+
19+
# Quick-start development settings - unsuitable for production
20+
# See https://docs.djangoproject.com/en/1.10/howto/deployment/checklist/
21+
22+
# SECURITY WARNING: keep the secret key used in production secret!
23+
SECRET_KEY = '-gsgh@v@@5h*_l1a$m_=d-e((w!0ny1v-z)5vt5rag33b88d*d'
24+
25+
# SECURITY WARNING: don't run with debug turned on in production!
26+
DEBUG = True
27+
28+
ALLOWED_HOSTS = []
29+
30+
31+
# Application definition
32+
33+
INSTALLED_APPS = [
34+
'django.contrib.admin',
35+
'django.contrib.auth',
36+
'django.contrib.contenttypes',
37+
'django.contrib.sessions',
38+
'django.contrib.messages',
39+
'django.contrib.staticfiles',
40+
]
41+
42+
MIDDLEWARE = [
43+
'django.middleware.security.SecurityMiddleware',
44+
'django.contrib.sessions.middleware.SessionMiddleware',
45+
'django.middleware.common.CommonMiddleware',
46+
'django.middleware.csrf.CsrfViewMiddleware',
47+
'django.contrib.auth.middleware.AuthenticationMiddleware',
48+
'django.contrib.messages.middleware.MessageMiddleware',
49+
'django.middleware.clickjacking.XFrameOptionsMiddleware',
50+
]
51+
52+
ROOT_URLCONF = 'cvApi.urls'
53+
54+
TEMPLATES = [
55+
{
56+
'BACKEND': 'django.template.backends.django.DjangoTemplates',
57+
'DIRS': [],
58+
'APP_DIRS': True,
59+
'OPTIONS': {
60+
'context_processors': [
61+
'django.template.context_processors.debug',
62+
'django.template.context_processors.request',
63+
'django.contrib.auth.context_processors.auth',
64+
'django.contrib.messages.context_processors.messages',
65+
],
66+
},
67+
},
68+
]
69+
70+
WSGI_APPLICATION = 'cvApi.wsgi.application'
71+
72+
73+
# Database
74+
# https://docs.djangoproject.com/en/1.10/ref/settings/#databases
75+
76+
DATABASES = {
77+
'default': {
78+
'ENGINE': 'django.db.backends.sqlite3',
79+
'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
80+
}
81+
}
82+
83+
84+
# Password validation
85+
# https://docs.djangoproject.com/en/1.10/ref/settings/#auth-password-validators
86+
87+
AUTH_PASSWORD_VALIDATORS = [
88+
{
89+
'NAME': 'django.contrib.auth.password_validation.UserAttributeSimilarityValidator',
90+
},
91+
{
92+
'NAME': 'django.contrib.auth.password_validation.MinimumLengthValidator',
93+
},
94+
{
95+
'NAME': 'django.contrib.auth.password_validation.CommonPasswordValidator',
96+
},
97+
{
98+
'NAME': 'django.contrib.auth.password_validation.NumericPasswordValidator',
99+
},
100+
]
101+
102+
103+
# Internationalization
104+
# https://docs.djangoproject.com/en/1.10/topics/i18n/
105+
106+
LANGUAGE_CODE = 'en-us'
107+
108+
TIME_ZONE = 'UTC'
109+
110+
USE_I18N = True
111+
112+
USE_L10N = True
113+
114+
USE_TZ = True
115+
116+
117+
# Static files (CSS, JavaScript, Images)
118+
# https://docs.djangoproject.com/en/1.10/howto/static-files/
119+
120+
STATIC_URL = '/static/'

cvApi/settings.pyc

2.51 KB
Binary file not shown.

cvApi/urls.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
"""cvApi 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, include
17+
from django.contrib import admin
18+
19+
urlpatterns = [
20+
url(r'^admin/', admin.site.urls),
21+
url(r'^face_detection/', include('face_detector.urls')),
22+
]

cvApi/urls.pyc

1.05 KB
Binary file not shown.

cvApi/wsgi.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
"""
2+
WSGI config for cvApi project.
3+
4+
It exposes the WSGI callable as a module-level variable named ``application``.
5+
6+
For more information on this file, see
7+
https://docs.djangoproject.com/en/1.10/howto/deployment/wsgi/
8+
"""
9+
10+
import os
11+
12+
from django.core.wsgi import get_wsgi_application
13+
14+
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "cvApi.settings")
15+
16+
application = get_wsgi_application()

cvApi/wsgi.pyc

603 Bytes
Binary file not shown.

db.sqlite3

128 KB
Binary file not shown.

face_detector/__init__.py

Whitespace-only changes.

face_detector/__init__.pyc

157 Bytes
Binary file not shown.

face_detector/admin.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from django.contrib import admin
2+
3+
# Register your models here.

face_detector/apps.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
from __future__ import unicode_literals
2+
3+
from django.apps import AppConfig
4+
5+
6+
class FaceDetectorConfig(AppConfig):
7+
name = 'face_detector'

face_detector/migrations/__init__.py

Whitespace-only changes.

face_detector/models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from __future__ import unicode_literals
2+
3+
from django.db import models
4+
5+
# Create your models here.

face_detector/tests.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from django.test import TestCase
2+
3+
# Create your tests here.

face_detector/urls.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
"""cvApi 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, include
17+
from django.contrib import admin
18+
from . import views
19+
urlpatterns = [
20+
url(r'^detect/', views.detect, name="detect"),
21+
]

face_detector/urls.pyc

1.06 KB
Binary file not shown.

face_detector/views.py

Lines changed: 77 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,77 @@
1+
# import the necessary packages
2+
from django.views.decorators.csrf import csrf_exempt
3+
from django.http import JsonResponse
4+
import numpy as np
5+
import urllib
6+
import json
7+
import cv2
8+
import os
9+
10+
# define the path to the face detector
11+
FACE_DETECTOR_PATH = "{base_path}/cascades/haarcascade_frontalface_default.xml".format(
12+
base_path=os.path.abspath(os.path.dirname(__file__)))
13+
14+
@csrf_exempt
15+
def detect(request):
16+
# initialize the data dictionary to be returned by the request
17+
data = {"success": False}
18+
19+
# check to see if this is a post request
20+
if request.method == "POST":
21+
# check to see if an image was uploaded
22+
if request.FILES.get("image", None) is not None:
23+
# grab the uploaded image
24+
image = _grab_image(stream=request.FILES["image"])
25+
26+
# otherwise, assume that a URL was passed in
27+
else:
28+
# grab the URL from the request
29+
url = request.POST.get("url", None)
30+
31+
# if the URL is None, then return an error
32+
if url is None:
33+
data["error"] = "No URL provided."
34+
return JsonResponse(data)
35+
36+
# load the image and convert
37+
image = _grab_image(url=url)
38+
39+
# convert the image to grayscale, load the face cascade detector,
40+
# and detect faces in the image
41+
image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
42+
detector = cv2.CascadeClassifier(FACE_DETECTOR_PATH)
43+
rects = detector.detectMultiScale(image, scaleFactor=1.1, minNeighbors=5,
44+
minSize=(30, 30), flags=cv2.cv.CV_HAAR_SCALE_IMAGE)
45+
46+
# construct a list of bounding boxes from the detection
47+
rects = [(int(x), int(y), int(x + w), int(y + h)) for (x, y, w, h) in rects]
48+
49+
# update the data dictionary with the faces detected
50+
data.update({"num_faces": len(rects), "faces": rects, "success": True})
51+
52+
# return a JSON response
53+
return JsonResponse(data)
54+
55+
def _grab_image(path=None, stream=None, url=None):
56+
# if the path is not None, then load the image from disk
57+
if path is not None:
58+
image = cv2.imread(path)
59+
60+
# otherwise, the image does not reside on disk
61+
else:
62+
# if the URL is not None, then download the image
63+
if url is not None:
64+
resp = urllib.urlopen(url)
65+
data = resp.read()
66+
67+
# if the stream is not None, then the image has been uploaded
68+
elif stream is not None:
69+
data = stream.read()
70+
71+
# convert the image to a NumPy array and then read it into
72+
# OpenCV format
73+
image = np.asarray(bytearray(data), dtype="uint8")
74+
image = cv2.imdecode(image, cv2.IMREAD_COLOR)
75+
76+
# return the image
77+
return image

face_detector/views.pyc

2.12 KB
Binary file not shown.

manage.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
#!/usr/bin/env python
2+
import os
3+
import sys
4+
5+
if __name__ == "__main__":
6+
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "cvApi.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)

obama.jpg

93.1 KB
Loading

test_api.py

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
# import the necessary packages
2+
import requests
3+
import cv2
4+
5+
# define the URL to our face detection API
6+
url = "http://localhost:8000/face_detection/detect/"
7+
8+
# use our face detection API to find faces in images via image URL
9+
image = cv2.imread("obama.jpg")
10+
payload = {"url": "http://www.pyimagesearch.com/wp-content/uploads/2015/05/obama.jpg"}
11+
r = requests.post(url, data=payload).json()
12+
print "obama.jpg: {}".format(r)
13+
14+
# loop over the faces and draw them on the image
15+
for (startX, startY, endX, endY) in r["faces"]:
16+
cv2.rectangle(image, (startX, startY), (endX, endY), (0, 255, 0), 2)
17+
18+
# show the output image
19+
cv2.imshow("obama.jpg", image)
20+
cv2.waitKey(0)
21+
22+
# load our image and now use the face detection API to find faces in
23+
# images by uploading an image directly
24+
image = cv2.imread("adrian.jpg")
25+
payload = {"image": open("adrian.jpg", "rb")}
26+
r = requests.post(url, files=payload).json()
27+
print "adrian.jpg: {}".format(r)
28+
29+
# loop over the faces and draw them on the image
30+
for (startX, startY, endX, endY) in r["faces"]:
31+
cv2.rectangle(image, (startX, startY), (endX, endY), (0, 255, 0), 2)
32+
33+
# show the output image
34+
cv2.imshow("adrian.jpg", image)
35+
cv2.waitKey(0)

0 commit comments

Comments
 (0)