|
import os |
|
import cv2 |
|
import numpy as np |
|
import base64 |
|
import face_manage.manage as db_manage |
|
from flask import Flask, render_template, request, jsonify |
|
from extract import GetImageInfo |
|
|
|
app = Flask(__name__) |
|
|
|
UPLOAD_FOLDER = os.path.basename('uploads') |
|
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER |
|
|
|
|
|
@app.route("/") |
|
def start_page(): |
|
print("Start") |
|
response = jsonify({"status": "Start"}) |
|
response.status_code = 200 |
|
response.headers["Content-Type"] = "application/json; charset=utf-8" |
|
return response |
|
|
|
|
|
@app.route("/enroll") |
|
def enroll(): |
|
file = request.files['image'] |
|
image = cv2.imdecode(np.fromstring(file.read(), np.uint8), cv2.IMREAD_UNCHANGED) |
|
|
|
db_manage.open_database(0) |
|
count, boxes, scores, landmarks, alignimgs, features = GetImageInfo(image, 5) |
|
|
|
for idx in range(0, count): |
|
db_manage.register_face('sample name', idx, boxes[idx], landmarks[idx], alignimgs[idx], features[idx]) |
|
|
|
|
|
|
|
response = jsonify({"status": "True"}) |
|
response.status_code = 200 |
|
response.headers["Content-Type"] = "application/json; charset=utf-8" |
|
return response |
|
|
|
|
|
@app.route("/delete/all") |
|
def delete_all(): |
|
db_manage.open_database(0) |
|
db_manage.clear_database() |
|
|
|
response = jsonify({"status": "True"}) |
|
response.status_code = 200 |
|
response.headers["Content-Type"] = "application/json; charset=utf-8" |
|
return response |
|
|
|
|
|
@app.route("/match11") |
|
def match_1_1(): |
|
file1 = request.files['image1'] |
|
file2 = request.files['image2'] |
|
|
|
image1 = cv2.imdecode(np.fromstring(file1.read(), np.uint8), cv2.IMREAD_UNCHANGED) |
|
image2 = cv2.imdecode(np.fromstring(file2.read(), np.uint8), cv2.IMREAD_UNCHANGED) |
|
|
|
count1, boxes1, scores1, landmarks1, alignimgs1, features1 = GetImageInfo(image1, 1) |
|
count2, boxes2, scores2, landmarks2, alignimgs2, features2 = GetImageInfo(image2, 1) |
|
|
|
if count1 != 0 and count2 != 0: |
|
sim = db_manage.get_similarity(features1[0], features2[0]) |
|
if sim > db_manage.threshold: |
|
result = True |
|
else: |
|
result = False |
|
|
|
response = jsonify({"status": result}) |
|
response.status_code = 200 |
|
response.headers["Content-Type"] = "application/json; charset=utf-8" |
|
return response |
|
|
|
|
|
@app.route("/match1n") |
|
def match_1_n(): |
|
file = request.files['image'] |
|
image = cv2.imdecode(np.fromstring(file.read(), np.uint8), cv2.IMREAD_UNCHANGED) |
|
|
|
result, filename, sub_index = False, None, -1 |
|
count, boxes, scores, landmarks, alignimgs, features = GetImageInfo(image, 1) |
|
|
|
for idx in range(count): |
|
id, fn, sub_id = db_manage.verify_face(features[idx]) |
|
if id != -1: |
|
result, filename, sub_index = True, fn, id |
|
|
|
response = jsonify({"status": result, "filename": filename, "subIndex": sub_index}) |
|
response.status_code = 200 |
|
response.headers["Content-Type"] = "application/json; charset=utf-8" |
|
return response |
|
|
