From 414b0bd8fb586dab3a1f55e38f052748b5f20522 Mon Sep 17 00:00:00 2001 From: fabrice Date: Thu, 13 Jun 2019 12:11:56 +0000 Subject: [PATCH] Update 'face_recognition/app/app.py' --- face_recognition/app/app.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/face_recognition/app/app.py b/face_recognition/app/app.py index f081569..aaa38bc 100644 --- a/face_recognition/app/app.py +++ b/face_recognition/app/app.py @@ -4,6 +4,7 @@ from minio.error import (ResponseError, BucketAlreadyOwnedByYou, BucketAlreadyEx import os import uuid import pickle +import face_recognition app = Flask(__name__) minioClient = Minio('minio:9000', access_key=os.environ['s3-name'], secret_key=os.environ['s3-password'], secure=False) @@ -36,6 +37,12 @@ def new_user_id(): def new_face(): id = request.args.get('id', None) upload_id = request.args.get('upload_id', None) + minioClient.fget_object('uploads', upload_id, '/tmp/'+str(upload_id)) + face = face_recognition.load_image_file('/tmp/'+str(upload_id)) + face_encoding = face_recognition.face_encodings(face)[0] + with open('/tmp/'+str(id)+'.pkl', 'wb') as file: + pickle.dump(face_encoding, file) + minioClient.fput_object('users', str(id), '/tmp/'+str(id)+'.pkl') return str(id) + str(upload_id) def check_id(id): @@ -59,6 +66,14 @@ def setup(): pass except ResponseError as err: raise + try: + minioClient.make_bucket("uploads") + except BucketAlreadyOwnedByYou as err: + pass + except BucketAlreadyExists as err: + pass + except ResponseError as err: + raise if __name__ == '__main__':