好友
阅读权限10
听众
最后登录1970-1-1
|
一、 需求分析选题“员工刷脸考勤”，要求采用python语言开发，可以通过摄像头添加员工面部信息，这里就涉及到两个具体的个问题，一个是应该以什么样的数据来标识每一个员工的面部信息，二是持久化地保存这些信息到数据库中去。更细地，还涉及表的设计；另一个基本要求是通过摄像头识别员工面部信息来完成考勤，这个问题基本可以通过遍历数据库里的员工面部数据与当前摄像头里的员工面部数据的比对来实现，但有一个问题就是假如摄像头里有多张人脸改怎么处理。扩展要求是导出每日的考勤表，可以拆分为两个部分，一个是存储考勤信息，一个是展示考勤信息。我们希望达到的目标是：（1）仿照通用型软件界面设计的原则，所有的操作都在菜单栏里实现，一部分区域用于展示摄像头实时读取并由程序加工后的视频流信息，另一部分区域做控制台输出，打印相关信息，比如提示员工面部信息添加成功、添加失败及其原因，提示员工打卡成功、打卡失败及其原因；添加面部信息时人是必须和程序进行交互的，比如输入一些相关的信息，这个时候程序是阻塞的；但是在打卡的时候，程序是不阻塞的，如果不点击关闭打卡，它会一直在打卡的模式，等待并识别每一个前来打卡的员工，这比较符合现实的使用场景。（2）建表来存储员工信息和考勤信息，每次新建录入员工面部信息时，要求输入工号、姓名，并查无重后方可录入，录入时只取距离屏幕最近的员工的面部信息，这是考虑到实际打卡都是依次进行而不是一群人一群人地打卡，录入时有两种模式可供选择，自动模式：一旦识别到人脸就自动捕获截图，连续截图达到10张就结束录入；手动模式：点击菜单结束录入，不一定要得到10张。结束录入后就开始从刚才捕获的员工面部提取人脸特征数据并连同之前输入的员工姓名等数据作为一行记录保存到数据库中，如果没有捕获到员工面部或者刚才捕获的面部信息不是同一个人，这行记录就被丢弃。 考勤时，须满足三个条件：面部信息已录入、在打卡时间段内、未重复打卡，只有打卡成功，打卡人姓名及工号、打卡日期及时间才会被当成一行记录保存到数据库并在控制台输出打卡成功信息，否则会在控制台输出失败及其原因信息。 总而言之：我们的设计目标是规范化、人性化。二、 总体设计为了完成上述目标一，程序的界面初始化分为三部分，第一部分初始化菜单栏，第二部分初始化左边控制台，第三部分初始化右边展示面板，使这三部分相互独立；数据逻辑部分的初始化分为两部分，第一部分是数据库部分的初始化，如果数据库/表不存在就新建，存在则加载相关数据，第二部分是初始化一些需要循环使用的变量，比如新建录入时的员工姓名、工号、截图数目计数器等,每当完成录入时这些数据都应该被重置成初始化以待下一次录入，把这些初始化语句写成一个函数可以提高代码复用度。 上述目标二主要是一些限制性条件，可以通过添加判断语句来实现，比如对输入id的合法性检验： whileself.id == ID_WORKER_UNAVIABLE: self.id =wx.GetNumberFromUser(message="请输入您的工号(-1不可用)",prompt="工号", caption="温馨提示",value=ID_WORKER_UNAVIABLE, parent=self.bmp,max=100000000,min=ID_WORKER_UNAVIABLE) for knew_id in self.knew_id: if knew_id == self.id: self.id =ID_WORKER_UNAVIABLE wx.MessageBox(message="工号已存在，请重新输入", caption="警告") 其中ID_WORKER_UNAVIABLE是id的初始化值-1，不可用，self.knew.id是从数据库里加载出来的id列表，如果id非法（已重复或者不在0~100000000，就会一直有新的弹窗来提示输入id。 再比如对拒绝多张人脸时、只处理距离屏幕最近的员工的面部信息： if len(dets) != 0: biggest_face = dets[0] #取占比最大的脸 maxArea = 0 for det in dets: w = det.right() -det.left() h = det.top()-det.bottom() if w*h > maxArea: biggest_face = det maxArea = w*h dets是侦测到的所有面部数组，biggest_face是距离屏幕最近的面部。 本次课设的完整过程如下：选了员工刷脸考勤系统以后，我和小组成员刘佳璇查阅了大量资料，了解到dlib这个库，于是我们利用dlib库自带的人脸预测器、特征提取器，然后计算128d特征来完成人脸识别，再加上我们之前构想的逻辑，大致完成了课设，由于完成得早，我们大致在课设安排时间的一半时就给老师检查，虽然基本要求和拓展要求都已经实现，但是还有许多可以完善的地方，老师给出了许多宝贵的意见：（1） 把数据保存的地方由csv文件改成数据库。（2） 减少弹出式界面，用frame替换的方式解决之。（3） 刷脸打卡的界面不需要打卡者交互，全自动打卡。最后检查的时候这些问题都被解决了。 程序框图： 注：图片如看不清也可在线预览：https://www.processon.com/view/link/5bbcc953e4b08faf8c7324a1 本程序的设计思想大致可分为以下几个方面(1)面向对象的原则，整个程序的主体就是一个WAS(WorkAttendanceSystem)类,所有的实现都围绕这个类展开。(2)界面和数据逻辑分离的原则，WAS类的初始化过程包括界面的初始化和数据初始化，两者相互独立。(3)代码封装原则，多次调用的语句集写成接口供调用，没有冗余的代码。(4)接口隔离原则:使用多个专门的接口,而不是使用单一的总接口。 三、 详细设计函数清单注：所有类内的函数的第一个参数为self，表明该函数属于该类，后面不再赘述 （1） def __init__(self)WAS类的构造函数，主要是完成一些初始化操作，如初始化菜单、信息打印面板、主展示面板以及初始化加载数据库、初始化循环使用的变量。 （2） definitMenu(self):完成菜单的初始化显示，点击事件绑定。（3） def initInfoText(self):完成左边信息提示面板的初始化显示。 （4） def initGallery(self):完成右边主展示面板的初始化显示。 （5） definitDatabase(self):数据库的初始化，建立数据库连接（如果数据库inspurer.db不存在则先新建），如果数据库中不存在员工信息worker_info和考勤logcat这两个表，则依次创建。 （6） defloadDataBase(self,type):该模块函数完成从数据库读取数据的操作，包括读取员工信息和考勤信息，第二个参数type用于标识是加载员工信息还是考勤信息，一方面，可以统一接口，打开数据库和得到游标、关闭连接是一样的，将两个读取接口合二为一，提高代码复用度；另一方面，可以减少加载的工作量，减少IO,提高程序运行速度；最后，因为读取信息前对上一次读取的信息列表做了清空处理，用type标识可以避免读取一个表时对另一个表造成的误操作。 （7） def insertARow(self,Row,type): 该模块函数完成写数据库操作，第二个参数为准备写的一条记录，第三个参数type表示要对哪一个表进行写操作。 （8） def adapt_array(self,arr):将提取的人脸特征信息（列表）压缩，入口参数就是待压缩的数据，出口参数是压缩后的数据，用于写入数据库。 （9） def convert_array(self,text):将读取出来的数据解压缩成人脸特征信息，入口参数是待解压得数据，出口参数是解压后的数据。 （10） def return_euclidean_distance(feature_1, feature_2): 计算两个人脸的欧式距离，入口参数是两个人脸的特征数据，出口参数是判定的结果，欧式距离大于0.4判为不同，不大于判为相同。 （11） def OnNewRegisterClicked(self,event):见名知义，菜单新建录入的监听事件，参数event为事件信息，其他几个菜单的(OnFinishRegisterClicked,OnStartPunchCardClicked, OnEndPunchCardClicked, OnOpenLogcatClicked,OnCloseLogcatClicked)类似，在此不再赘述。 （12） def getDateAndTime(self):得到当前日期和时间，并组装成特定格式作为出口参数返回。
函数调用关系:箭头指向被调用者
一、 代码#coding=utf-8import wximport wx.gridimport sqlite3from time import localtime,strftimeimport osfrom skimage import io as iioimport ioimport zlibimport dlib # 人脸识别的库dlibimport numpy as np # 数据处理的库numpyimport cv2 # 图像处理的库OpenCvimport _thread ID_NEW_REGISTER = 160ID_FINISH_REGISTER = 161 ID_START_PUNCHCARD = 190ID_END_PUNCARD = 191 ID_OPEN_LOGCAT = 283ID_CLOSE_LOGCAT = 284 ID_WORKER_UNAVIABLE = -1 PATH_FACE ="data/face_img_database/"# face recognition model, the object mapshuman faces into 128D vectorsfacerec =dlib.face_recognition_model_v1("model/dlib_face_recognition_resnet_model_v1.dat")# Dlib 预测器detector = dlib.get_frontal_face_detector()predictor =dlib.shape_predictor('model/shape_predictor_68_face_landmarks.dat')def return_euclidean_distance(feature_1,feature_2): feature_1 = np.array(feature_1) feature_2 = np.array(feature_2) dist = np.sqrt(np.sum(np.square(feature_1 - feature_2))) print("欧式距离: ", dist) if dist > 0.4: return "diff" else: return "same" class WAS(wx.Frame): def __init__(self): wx.Frame.__init__(self,parent=None,title="员工考勤系统",size=(920,560)) self.initMenu() self.initInfoText() self.initGallery() self.initDatabase() self.initData() def initData(self): self.name = "" self.id =ID_WORKER_UNAVIABLE self.face_feature = "" self.pic_num = 0 self.flag_registed = False self.puncard_time = "09:00:00" self.loadDataBase(1) def initMenu(self): menuBar = wx.MenuBar() #生成菜单栏 menu_Font = wx.Font()#Font(faceName="consolas",pointsize=20) menu_Font.SetPointSize(14) menu_Font.SetWeight(wx.BOLD) registerMenu = wx.Menu() #生成菜单 self.new_register = wx.MenuItem(registerMenu,ID_NEW_REGISTER,"新建录入") self.new_register.SetBitmap(wx.Bitmap("drawable/new_register.png")) self.new_register.SetTextColour("SLATE BLUE") self.new_register.SetFont(menu_Font) registerMenu.Append(self.new_register) self.finish_register =wx.MenuItem(registerMenu,ID_FINISH_REGISTER,"完成录入") self.finish_register.SetBitmap(wx.Bitmap("drawable/finish_register.png")) self.finish_register.SetTextColour("SLATE BLUE") self.finish_register.SetFont(menu_Font) self.finish_register.Enable(False) registerMenu.Append(self.finish_register) puncardMenu = wx.Menu() self.start_punchcard = wx.MenuItem(puncardMenu,ID_START_PUNCHCARD,"开始签到") self.start_punchcard.SetBitmap(wx.Bitmap("drawable/start_punchcard.png")) self.start_punchcard.SetTextColour("SLATE BLUE") self.start_punchcard.SetFont(menu_Font) puncardMenu.Append(self.start_punchcard) self.end_puncard = wx.MenuItem(puncardMenu,ID_END_PUNCARD,"结束签到") self.end_puncard.SetBitmap(wx.Bitmap("drawable/end_puncard.png")) self.end_puncard.SetTextColour("SLATE BLUE") self.end_puncard.SetFont(menu_Font) self.end_puncard.Enable(False) puncardMenu.Append(self.end_puncard) logcatMenu = wx.Menu() self.open_logcat = wx.MenuItem(logcatMenu,ID_OPEN_LOGCAT,"打开日志") self.open_logcat.SetBitmap(wx.Bitmap("drawable/open_logcat.png")) self.open_logcat.SetFont(menu_Font) self.open_logcat.SetTextColour("SLATE BLUE") logcatMenu.Append(self.open_logcat) self.close_logcat = wx.MenuItem(logcatMenu, ID_CLOSE_LOGCAT, "关闭日志") self.close_logcat.SetBitmap(wx.Bitmap("drawable/close_logcat.png")) self.close_logcat.SetFont(menu_Font) self.close_logcat.SetTextColour("SLATE BLUE") logcatMenu.Append(self.close_logcat) menuBar.Append(registerMenu,"&人脸录入") menuBar.Append(puncardMenu,"&刷脸签到") menuBar.Append(logcatMenu,"&考勤日志") self.SetMenuBar(menuBar) self.Bind(wx.EVT_MENU,self.OnNewRegisterClicked,id=ID_NEW_REGISTER) self.Bind(wx.EVT_MENU,self.OnFinishRegisterClicked,id=ID_FINISH_REGISTER) self.Bind(wx.EVT_MENU,self.OnStartPunchCardClicked,id=ID_START_PUNCHCARD) self.Bind(wx.EVT_MENU,self.OnEndPunchCardClicked,id=ID_END_PUNCARD) self.Bind(wx.EVT_MENU,self.OnOpenLogcatClicked,id=ID_OPEN_LOGCAT) self.Bind(wx.EVT_MENU,self.OnCloseLogcatClicked,id=ID_CLOSE_LOGCAT) def OnOpenLogcatClicked(self,event): self.loadDataBase(2) grid = wx.grid.Grid(self,pos=(320,0),size=(600,500)) grid.CreateGrid(100, 4) for i in range(100): for j in range(4): grid.SetCellAlignment(i,j,wx.ALIGN_CENTER,wx.ALIGN_CENTER) grid.SetColLabelValue(0, "工号") #第一列标签 grid.SetColLabelValue(1, "姓名") grid.SetColLabelValue(2, "打卡时间") grid.SetColLabelValue(3, "是否迟到") grid.SetColSize(0,100) grid.SetColSize(1,100) grid.SetColSize(2,150) grid.SetColSize(3,150) grid.SetCellTextColour("NAVY") for i,id in enumerate(self.logcat_id): grid.SetCellValue(i,0,str(id)) grid.SetCellValue(i,1,self.logcat_name) grid.SetCellValue(i,2,self.logcat_datetime) grid.SetCellValue(i,3,self.logcat_late) pass def OnCloseLogcatClicked(self,event): self.initGallery() pass def register_cap(self,event): # 创建 cv2 摄像头对象 self.cap = cv2.VideoCapture(0) # cap.set(propId, value) # 设置视频参数，propId设置的视频参数，value设置的参数值 # self.cap.set(3, 600) # self.cap.set(4,600) # cap是否初始化成功 while self.cap.isOpened(): # cap.read() # 返回两个值： # 一个布尔值true/false，用来判断读取视频是否成功/是否到视频末尾 # 图像对象，图像的三维矩阵 flag, im_rd = self.cap.read() # 每帧数据延时1ms，延时为0读取的是静态帧 kk = cv2.waitKey(1) # 人脸数 dets dets = detector(im_rd, 1) # 检测到人脸 if len(dets) != 0: biggest_face = dets[0] #取占比最大的脸 maxArea = 0 for det in dets: w = det.right() -det.left() h = det.top()-det.bottom() if w*h > maxArea: biggest_face = det maxArea = w*h # 绘制矩形框 cv2.rectangle(im_rd,tuple([biggest_face.left(), biggest_face.top()]), tuple([biggest_face.right(), biggest_face.bottom()]), (255, 0,0), 2) img_height, img_width =im_rd.shape[:2] image1 = cv2.cvtColor(im_rd,cv2.COLOR_BGR2RGB) pic =wx.Bitmap.FromBuffer(img_width, img_height, image1) # 显示图片在panel上 self.bmp.SetBitmap(pic) # 获取当前捕获到的图像的所有人脸的特征，存储到features_cap_arr shape = predictor(im_rd,biggest_face) features_cap =facerec.compute_face_descriptor(im_rd, shape) # 对于某张人脸，遍历所有存储的人脸特征 for i,knew_face_feature inenumerate(self.knew_face_feature): # 将某张人脸与存储的所有人脸数据进行比对 compare =return_euclidean_distance(features_cap, knew_face_feature) if compare =="same": # 找到了相似脸 self.infoText.AppendText(self.getDateAndTime()+"工号:"+str(self.knew_id) +" 姓名:"+self.knew_name+"的人脸数据已存在\r\n") self.flag_registed =True self.OnFinishRegister() _thread.exit() #print(features_known_arr[-1]) face_height =biggest_face.bottom()-biggest_face.top() face_width =biggest_face.right()- biggest_face.left() im_blank =np.zeros((face_height, face_width, 3), np.uint8) try: for ii inrange(face_height): for jj inrange(face_width): im_blank[ii][jj] =im_rd[biggest_face.top() + ii][biggest_face.left() + jj] #cv2.imwrite(path_make_dir+self.name + "/img_face_" +str(self.sc_number) + ".jpg", im_blank) # cap =cv2.VideoCapture("***.mp4") #cap.set(cv2.CAP_PROP_POS_FRAMES, 2) # ret, frame = cap.read() # cv2.imwrite("我//h.jpg", frame) # 该方法不成功 # 解决python3下使用cv2.imwrite存储带有中文路径图片 if len(self.name)>0: cv2.imencode('.jpg',im_blank)[1].tofile( PATH_FACE + self.name +"/img_face_" + str(self.pic_num) + ".jpg") # 正确方法 self.pic_num += 1 print("写入本地：", str(PATH_FACE +self.name) + "/img_face_" + str(self.pic_num) + ".jpg") self.infoText.AppendText(self.getDateAndTime()+"图片:"+str(PATH_FACE + self.name)+ "/img_face_" + str(self.pic_num) + ".jpg保存成功\r\n") except: print("保存照片异常,请对准摄像头") if self.new_register.IsEnabled(): _thread.exit() if self.pic_num == 10: self.OnFinishRegister() _thread.exit() def OnNewRegisterClicked(self,event): self.new_register.Enable(False) self.finish_register.Enable(True) self.loadDataBase(1) while self.id == ID_WORKER_UNAVIABLE: self.id = wx.GetNumberFromUser(message="请输入您的工号(-1不可用)", prompt="工号", caption="温馨提示", value=ID_WORKER_UNAVIABLE, parent=self.bmp,max=100000000,min=ID_WORKER_UNAVIABLE) for knew_id in self.knew_id: if knew_id == self.id: self.id =ID_WORKER_UNAVIABLE wx.MessageBox(message="工号已存在，请重新输入", caption="警告") while self.name == '': self.name = wx.GetTextFromUser(message="请输入您的的姓名,用于创建姓名文件夹", caption="温馨提示", default_value="", parent=self.bmp) # 监测是否重名 for exsit_name in (os.listdir(PATH_FACE)): if self.name == exsit_name: wx.MessageBox(message="姓名文件夹已存在，请重新输入", caption="警告") self.name = '' break os.makedirs(PATH_FACE+self.name) _thread.start_new_thread(self.register_cap,(event,)) pass def OnFinishRegister(self): self.new_register.Enable(True) self.finish_register.Enable(False) self.cap.release() self.bmp.SetBitmap(wx.Bitmap(self.pic_index)) if self.flag_registed == True: dir = PATH_FACE + self.name for file in os.listdir(dir): os.remove(dir+"/"+file) print("已删除已录入人脸的图片", dir+"/"+file) os.rmdir(PATH_FACE + self.name) print("已删除已录入人脸的姓名文件夹", dir) self.initData() return if self.pic_num>0: pics = os.listdir(PATH_FACE + self.name) feature_list = [] feature_average = [] for i in range(len(pics)): pic_path = PATH_FACE +self.name + "/" + pics print("正在读的人脸图像：", pic_path) img = iio.imread(pic_path) img_gray =cv2.cvtColor(img, cv2.COLOR_BGR2RGB) dets = detector(img_gray, 1) if len(dets) != 0: shape = predictor(img_gray,dets[0]) face_descriptor = facerec.compute_face_descriptor(img_gray,shape) feature_list.append(face_descriptor) else: face_descriptor = 0 print("未在照片中识别到人脸") if len(feature_list) > 0: for j in range(128): #防止越界 feature_average.append(0) for i inrange(len(feature_list)): feature_average[j] +=feature_list[j] feature_average[j] =(feature_average[j]) / len(feature_list) self.insertARow([self.id,self.name,feature_average],1) self.infoText.AppendText(self.getDateAndTime()+"工号:"+str(self.id) +" 姓名:"+self.name+" 的人脸数据已成功存入\r\n") pass else: os.rmdir(PATH_FACE + self.name) print("已删除空文件夹",PATH_FACE + self.name) self.initData() def OnFinishRegisterClicked(self,event): self.OnFinishRegister() pass def punchcard_cap(self,event): self.cap = cv2.VideoCapture(0) # cap.set(propId, value) # 设置视频参数，propId设置的视频参数，value设置的参数值 # self.cap.set(3, 600) # self.cap.set(4,600) # cap是否初始化成功 while self.cap.isOpened(): # cap.read() # 返回两个值： # 一个布尔值true/false，用来判断读取视频是否成功/是否到视频末尾 # 图像对象，图像的三维矩阵 flag, im_rd = self.cap.read() # 每帧数据延时1ms，延时为0读取的是静态帧 kk = cv2.waitKey(1) # 人脸数 dets dets = detector(im_rd, 1) # 检测到人脸 if len(dets) != 0: biggest_face = dets[0] # 取占比最大的脸 maxArea = 0 for det in dets: w = det.right() -det.left() h = det.top() -det.bottom() if w * h > maxArea: biggest_face = det maxArea = w * h # 绘制矩形框 cv2.rectangle(im_rd,tuple([biggest_face.left(), biggest_face.top()]), tuple([biggest_face.right(), biggest_face.bottom()]), (255, 0, 255), 2) img_height, img_width =im_rd.shape[:2] image1 = cv2.cvtColor(im_rd,cv2.COLOR_BGR2RGB) pic =wx.Bitmap.FromBuffer(img_width, img_height, image1) # 显示图片在panel上 self.bmp.SetBitmap(pic) # 获取当前捕获到的图像的所有人脸的特征，存储到features_cap_arr shape = predictor(im_rd,biggest_face) features_cap =facerec.compute_face_descriptor(im_rd, shape) # 对于某张人脸，遍历所有存储的人脸特征 for i, knew_face_feature inenumerate(self.knew_face_feature): # 将某张人脸与存储的所有人脸数据进行比对 compare =return_euclidean_distance(features_cap, knew_face_feature) if compare == "same": # 找到了相似脸 print("same") flag = 0 nowdt =self.getDateAndTime() for j,logcat_name inenumerate(self.logcat_name): if logcat_name == self.knew_name and nowdt[0:nowdt.index(" ")] ==self.logcat_datetime[j][0:self.logcat_datetime[j].index(" ")]: self.infoText.AppendText(nowdt+"工号:"+ str(self.knew_id) + " 姓名:" + self.knew_name +" 签到失败,重复签到\r\n") flag = 1 break if flag == 1: break ifnowdt[nowdt.index(" ")+1:-1] <= self.puncard_time: self.infoText.AppendText(nowdt + "工号:" + str(self.knew_id) + " 姓名:" +self.knew_name + " 成功签到,且未迟到\r\n") self.insertARow([self.knew_id,self.knew_name,nowdt,"否"],1) else: self.infoText.AppendText(nowdt + "工号:" + str(self.knew_id) + " 姓名:" +self.knew_name + " 成功签到,但迟到了\r\n") self.insertARow([self.knew_id, self.knew_name, nowdt, "是"], 2) self.loadDataBase(2) break ifself.start_punchcard.IsEnabled(): self.bmp.SetBitmap(wx.Bitmap(self.pic_index)) _thread.exit() def OnStartPunchCardClicked(self,event): # cur_hour = datetime.datetime.now().hour # print(cur_hour) # if cur_hour>=8 or cur_hour<6: # wx.MessageBox(message='''您错过了今天的签到时间，请明天再来\n # 每天的签到时间是:6:00~7:59''',caption="警告") # return self.start_punchcard.Enable(False) self.end_puncard.Enable(True) self.loadDataBase(2) _thread.start_new_thread(self.punchcard_cap,(event,)) pass def OnEndPunchCardClicked(self,event): self.start_punchcard.Enable(True) self.end_puncard.Enable(False) pass def initInfoText(self): #少了这两句infoText背景颜色设置失败，莫名奇怪 resultText = wx.StaticText(parent=self, pos = (10,20),size=(90, 60)) resultText.SetBackgroundColour('red') self.info = "\r\n"+self.getDateAndTime()+"程序初始化成功\r\n" #第二个参数水平混动条 self.infoText = wx.TextCtrl(parent=self,size=(320,500), style=(wx.TE_MULTILINE|wx.HSCROLL|wx.TE_READONLY)) #前景色，也就是字体颜色 self.infoText.SetForegroundColour("ORANGE") self.infoText.SetLabel(self.info) #API:https://www.cnblogs.com/wangjian8888/p/6028777.html # 没有这样的重载函数造成"paris not a key word",只好Set font = wx.Font() font.SetPointSize(12) font.SetWeight(wx.BOLD) font.SetUnderlined(True) self.infoText.SetFont(font) self.infoText.SetBackgroundColour('TURQUOISE') pass def initGallery(self): self.pic_index = wx.Image("drawable/index.png", wx.BITMAP_TYPE_ANY).Scale(600,500) self.bmp = wx.StaticBitmap(parent=self, pos=(320,0),bitmap=wx.Bitmap(self.pic_index)) pass def getDateAndTime(self): dateandtime = strftime("%Y-%m-%d %H:%M:%S",localtime()) return "["+dateandtime+"]" #数据库部分 #初始化数据库 def initDatabase(self): conn = sqlite3.connect("inspurer.db") #建立数据库连接 cur = conn.cursor() #得到游标对象 cur.execute('''create table if not exists worker_info (name text not null, id int not null primary key, face_feature array not null)''') cur.execute('''create table if not exists logcat (datetime text not null, id int not null, name text not null, late text not null)''') cur.close() conn.commit() conn.close() def adapt_array(self,arr): out = io.BytesIO() np.save(out, arr) out.seek(0) dataa = out.read() # 压缩数据流 return sqlite3.Binary(zlib.compress(dataa, zlib.Z_BEST_COMPRESSION)) def convert_array(self,text): out = io.BytesIO(text) out.seek(0) dataa = out.read() # 解压缩数据流 out = io.BytesIO(zlib.decompress(dataa)) return np.load(out) def insertARow(self,Row,type): conn = sqlite3.connect("inspurer.db") # 建立数据库连接 cur = conn.cursor() # 得到游标对象 if type == 1: cur.execute("insert into worker_info (id,name,face_feature) values(?,?,?)", (Row[0],Row[1],self.adapt_array(Row[2]))) print("写人脸数据成功") if type == 2: cur.execute("insert into logcat (id,name,datetime,late)values(?,?,?,?)", (Row[0],Row[1],Row[2],Row[3])) print("写日志成功") pass cur.close() conn.commit() conn.close() pass def loadDataBase(self,type): conn = sqlite3.connect("inspurer.db") # 建立数据库连接 cur = conn.cursor() # 得到游标对象 if type == 1: self.knew_id = [] self.knew_name = [] self.knew_face_feature = [] cur.execute('select id,name,face_feature from worker_info') origin = cur.fetchall() for row in origin: print(row[0]) self.knew_id.append(row[0]) print(row[1]) self.knew_name.append(row[1]) print(self.convert_array(row[2])) self.knew_face_feature.append(self.convert_array(row[2])) if type == 2: self.logcat_id = [] self.logcat_name = [] self.logcat_datetime = [] self.logcat_late = [] cur.execute('select id,name,datetime,late from logcat') origin = cur.fetchall() for row in origin: print(row[0]) self.logcat_id.append(row[0]) print(row[1]) self.logcat_name.append(row[1]) print(row[2]) self.logcat_datetime.append(row[2]) print(row[3]) self.logcat_late.append(row[3]) passapp = wx.App()frame = WAS()frame.Show()app.MainLoop() |
-
详情看附件
免费评分
-
查看全部评分
|
发表于 2019-11-26 18:08
|
发表于 2019-11-26 18:11
发表于 2019-11-26 18:15
