From 62ca3cefbd5c4c539274bd2af354e03151a28f7d Mon Sep 17 00:00:00 2001 From: caoqianming Date: Thu, 24 Feb 2022 20:59:40 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BA=BA=E8=84=B8=E8=AF=86=E5=88=AB=E8=B0=83?= =?UTF-8?q?=E6=95=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- hb_server/apps/hrm/services.py | 8 ++++---- hb_server/server/urls.py | 16 ++++++++++------ 2 files changed, 14 insertions(+), 10 deletions(-) diff --git a/hb_server/apps/hrm/services.py b/hb_server/apps/hrm/services.py index 1abfb19..9a15522 100644 --- a/hb_server/apps/hrm/services.py +++ b/hb_server/apps/hrm/services.py @@ -31,22 +31,22 @@ class HRMService: face_users = cache.get('face_users') try: results = face_recognition.compare_faces(face_datas, - unknown_face_encoding, tolerance=0.4) + unknown_face_encoding, tolerance=0.45) except: - return None, '人脸未匹配' + return None, '人脸未匹配1' for index, value in enumerate(results): if value: # 识别成功 user = User.objects.get(id=face_users[index]) return user, '' - return None, '人脸未匹配' + return None, '人脸未匹配2' @classmethod def get_facedata_from_img(cls, img_path): try: photo_path = settings.BASE_DIR + img_path picture_of_me = face_recognition.load_image_file(photo_path) - my_face_encoding = face_recognition.face_encodings(picture_of_me, num_jitters=10)[0] + my_face_encoding = face_recognition.face_encodings(picture_of_me)[0] face_data_list = my_face_encoding.tolist() return face_data_list, '' except: diff --git a/hb_server/server/urls.py b/hb_server/server/urls.py index 7776b06..0ed07c3 100644 --- a/hb_server/server/urls.py +++ b/hb_server/server/urls.py @@ -25,21 +25,25 @@ from rest_framework import routers from rest_framework.documentation import include_docs_urls from django.views.generic import TemplateView from utils.view import GenSignature, UpdateDevelop - +import os router = routers.DefaultRouter() router.register('', FileViewSet, basename="file") -schema_view = get_schema_view( - openapi.Info( +schema_dict = dict( + info=openapi.Info( title="航玻ERP API", default_version='v1', contact=openapi.Contact(email="caoqianming@foxmail.com"), license=openapi.License(name="MIT License"), ), public=True, - permission_classes=[], - url="http://49.232.14.174:2222/" -) + permission_classes=[],) + +if os.getenv('DJANGO_SETTINGS_MODULE') != 'server.settings_dev': + schema_dict['url'] = "http://49.232.14.174:2222/" + +schema_view = get_schema_view(**schema_dict) + urlpatterns = [ path('api/admin/doc/', include('django.contrib.admindocs.urls')),