from django.shortcuts import render
#导包
from django.http import HttpResponse,HttpResponseRedirect
#导入类路由库
from django.views import View
#导入数据库类
from myapp.models import User
import json
#导入时间模块
import time
import os
from mymac.settings import UPLOAD_ROOT
import cv2
import numpy as np
from myapp.my_compare import main_my,parse_arguments
from django.utils.decorators import method_decorator
from django.db import connection
#定义注册逻辑
class Sign_Up(View):
#定义注册接收方法
def post(self,request):
username = request.POST.get(‘username‘)
cap = cv2.VideoCapture(0)
face_cascade = cv2.CascadeClassifier("C:/Users/lenvov/AppData/Local/Programs/Python/Python37/Lib/site-packages/cv2/data/haarcascade_frontalface_default.xml")
my_count = 1
while(1):
# 获取摄像头屏幕对象
ret,frame = cap.read()
gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY )
# 识别脸部
faces = face_cascade.detectMultiScale(gray,scaleFactor=1.2,minNeighbors=3,minSize=(32,32))
for facerect in faces:
# 框处人脸
x,y,w,h = facerect
cv2.rectangle(gray,(x-10,y-10),(x+w+10,y+h+10),(0,255,0),3)
f = cv2.resize(gray[y:(y+h),x:(x+w)],(200,200))
# 生成摄像头视角窗口
cv2.imshow("capture",gray)
# 等待结束 如果按q键就结束
if cv2.waitKey(1) & 0xff == ord(‘q‘):
cv2.imwrite(UPLOAD_ROOT+‘/‘+username+‘_face.jpg‘,f)
break
# 释放摄像头
cap.release()
cv2.destroyAllWindows()
res = User.objects.filter(username=username).count()
if res > 0:
return HttpResponse(‘请换一个用户名‘)
else:
user = User(username=username)
user.save()
return HttpResponse(‘注册成功‘)
#定义登录逻辑
class Log_IN(View):
#定义注册接收方法
def post(self,request):
username = request.POST.get(‘username‘)
cap = cv2.VideoCapture(0)
face_cascade = cv2.CascadeClassifier("C:/Users/lenvov/AppData/Local/Programs/Python/Python37/Lib/site-packages/cv2/data/haarcascade_frontalface_default.xml")
my_count = 1
while(1):
# 获取摄像头屏幕对象
ret,frame = cap.read()
gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY )
# 识别脸部
faces = face_cascade.detectMultiScale(gray,scaleFactor=1.2,minNeighbors=3,minSize=(32,32))
for facerect in faces:
# 框处人脸
x,y,w,h = facerect
cv2.rectangle(gray,(x-10,y-10),(x+w+10,y+h+10),(0,255,0),3)
f = cv2.resize(gray[y:(y+h),x:(x+w)],(200,200))
# 生成摄像头视角窗口
cv2.imshow("capture",gray)
# 等待结束 如果按q键就结束
if cv2.waitKey(1) & 0xff == ord(‘q‘):
cv2.imwrite(UPLOAD_ROOT+‘/‘+username+‘_hou.jpg‘,f)
break
# 释放摄像头
cap.release()
cv2.destroyAllWindows()
res = User.objects.filter(username=username)[0]
args_list = []
args_list.append(‘C:/20170512-110547‘)
args_list.append(UPLOAD_ROOT+‘/‘+username+‘_face.jpg‘)
args_list.append(UPLOAD_ROOT+‘/‘+username+‘_hou.jpg‘)
res_float = main_my(parse_arguments(args_list))
print(res_float)
if res_float > 0.50:
return HttpResponse(‘识别头像不匹配‘)
else:
return HttpResponse(‘登录成功‘)
这些只是后端代码，前端代码就是通过axios来实现的。
如需有有疑问，可以在评论区里评论。
