import numpy as np??# 數據處理的庫 Numpy

import cv2? ?? ?? ? # 圖像處理的庫 OpenCv

import os

import shutil

import _thread

import wx

import csv

from importlib import reload

from skimage import io as iio

import face_recognize_punchcard

import sys

# 創建 cv2 攝像頭對象

#? ? C++: VideoCapture::VideoCapture(int device);

#API:http://www.opencv.org.cn/opencvdoc/2.3.2/html/modules/highgui/doc/reading_and_writing_images_and_video.html#videocapture

# 保存

path_make_dir = "data/face_img_database/"

path_feature_all = "data/feature_all.csv"

info = 'icon/info.png'

#register ui

class? ?RegisterUi(wx.Frame):

def __init__(self,superion):

wx.Frame.__init__(self,parent=superion,title="人臉錄入",size=(800,590),style=wx.DEFAULT_FRAME_STYLE|wx.STAY_ON_TOP)

self.SetBackgroundColour('white')

self.Center()

self.NewButton =??wx.Button(parent=self,pos=(50,120),size=(80,50),label='新建錄入')

self.ShortCutButton = wx.Button(parent=self,pos=(50,220),size=(80,50),label='截圖保存')

self.SaveButton =??wx.Button(parent=self,pos=(50,320),size=(80,50),label='完成錄入')

# 封面圖片

self.image_info = wx.Image(info, wx.BITMAP_TYPE_ANY).Scale(600, 480)

# 顯示圖片

self.bmp = wx.StaticBitmap(parent=self, pos=(180,20), bitmap=wx.Bitmap(self.image_info))

self.Bind(wx.EVT_BUTTON,self.OnShortCutButtonClicked,self.ShortCutButton)

self.Bind(wx.EVT_BUTTON,self.OnNewButtonClicked,self.NewButton)

self.ShortCutButton.Enable(enable=False)

self.SaveButton.Enable(False)

self.Bind(wx.EVT_BUTTON,self.OnSaveButtonClicked,self.SaveButton)

self.sc_number = 0

self.register_flag = 0

self.name = ""

def OnNewButtonClicked(self, event):

while self.name == '':

self.name = wx.GetTextFromUser(message="請先輸入錄入者的姓名,用于創建姓名文件夾", caption="溫馨提示",

default_value="", parent=None)

# 監測是否重名

for exsit_name in (os.listdir(path_make_dir)):

if self.name == exsit_name:

wx.MessageBox(message="姓名已存在，請重新輸入", caption="警告")

self.name = ''

break

os.makedirs(path_make_dir+self.name)

print("新建的人臉文件夾: ", path_make_dir+self.name)

self.NewButton.Enable(enable=False)

self.ShortCutButton.Enable(enable=True)

"""使用多線程，子線程運行后臺的程序，主線程更新前臺的UI，這樣不會互相影響"""

# 創建子線程，按鈕調用這個方法，

_thread.start_new_thread(self._open_cap, (event,))

def OnShortCutButtonClicked(self,event):

self.SaveButton.Enable(True)

if len(self.rects) !=0:

# 計算矩形框大小,保證同步

height = self.rects[0].bottom() - self.rects[0].top()

width = self.rects[0].right() - self.rects[0].left()

self.sc_number += 1

im_blank = np.zeros((height, width, 3), np.uint8)

for ii in range(height):

for jj in range(width):

im_blank[ii][jj] = self.im_rd[self.rects[0].top() + ii][self.rects[0].left() + jj]

cv2.imencode('.jpg', im_blank)[1].tofile(path_make_dir+self.name + "/img_face_" + str(self.sc_number) + ".jpg") #正確方法

print("寫入本地：", str(path_make_dir+self.name) + "/img_face_" + str(self.sc_number) + ".jpg")

else:

print("未檢測到人臉，識別無效，未寫入本地")

def OnSaveButtonClicked(self,event):

self.bmp.SetBitmap(wx.Bitmap(self.image_info))

self.NewButton.Enable(True)

self.SaveButton.Enable(False)

self.ShortCutButton.Enable(False)

# 釋放攝像頭

self.cap.release()

# 刪除建立的窗口

#cv2.destroyAllWindows()

if self.register_flag == 1:

if os.path.exists(path_make_dir+self.name):

shutil.rmtree(path_make_dir+self.name)

print("重復錄入，已刪除姓名文件夾", path_make_dir+self.name)

if self.sc_number == 0 and len(self.name)>0:

if os.path.exists(path_make_dir+self.name):

shutil.rmtree(path_make_dir+self.name)

print("您未保存截圖，已刪除姓名文件夾", path_make_dir+self.name)

if self.register_flag==0 and self.sc_number!=0:

pics = os.listdir(path_make_dir+self.name)

feature_list = []

feature_average = []

for i in range(len(pics)):

pic_path = path_make_dir+self.name + "/" + pics[i]

print("正在讀的人臉圖像：", pic_path)

img = iio.imread(pic_path)

img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

dets = face_recognize_punchcard.detector(img_gray, 1)

if len(dets) != 0:

shape = face_recognize_punchcard.predictor(img_gray, dets[0])

face_descriptor = face_recognize_punchcard.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 in range(len(feature_list)):

feature_average[j] += feature_list[i][j]

feature_average[j] = (feature_average[j])/len(feature_list)

feature_average.append(self.name)

with open(path_feature_all, "a+", newline="") as csvfile:

writer = csv.writer(csvfile)

print('寫入一條特征人臉入庫',feature_average)

writer.writerow(feature_average)

self.name = ""

self.register_flag = 0

self.sc_number = 0;

def _open_cap(self,event):

reload(face_recognize_punchcard)

urlc = 'http://192.168.137.119:8081/'

self.cap = cv2.VideoCapture(urlc)

# cap.set(propId, value)

# 設置視頻參數，propId 設置的視頻參數，value 設置的參數值

self.cap.set(3, 480)

#self.cap.set(cv2.CAP_PROP_FPS,5)

while self.cap.isOpened():

# cap.read()

# 返回兩個值：

#? ? 一個布爾值 true/false，用來判斷讀取視頻是否成功/是否到視頻末尾

#? ? 圖像對象，圖像的三維矩陣q

flag, self.im_rd = self.cap.read()

# 人臉數 rects

self.rects = face_recognize_punchcard.detector(self.im_rd, 1)

cv2.waitKey(1)??# 必不可少

# 待會要寫的字體

font = cv2.FONT_HERSHEY_SIMPLEX

if len(self.rects) != 0:

# 檢測到人臉

# 矩形框#d是人臉

# 查重

features_cap = face_recognize_punchcard.facerec.compute_face_descriptor(self.im_rd, face_recognize_punchcard.predictor(self.im_rd, self.rects[0]))

for i in range(len(face_recognize_punchcard.features_known_arr)):

# 將某張人臉與存儲的所有人臉數據進行比對

compare = face_recognize_punchcard.return_euclidean_distance(features_cap, face_recognize_punchcard.features_known_arr[i][0:-1])

if compare == "same":??# 找到了相似臉

face_name = face_recognize_punchcard.features_known_arr[i][-1]

print(face_name)

wx.MessageBox(message=face_name + "，您已錄過人臉，請檢查是否簽過到", caption="警告")

self.NewButton.Enable(False)

self.ShortCutButton.Enable(False)

self.SaveButton.Enable(True)

self.register_flag = 1

for k, d in enumerate(self.rects):

# 根據人臉大小生成空的圖像

# 最后一個參數是線寬

cv2.rectangle(self.im_rd, tuple([d.left(), d.top()]), tuple([d.right(), d.bottom()]), (255, 0, 0), 2)

# 顯示人臉數

cv2.putText(self.im_rd, "Faces: " + str(len(self.rects)), (50, 80), font, 0.8, (255, 0, 0), 1, cv2.LINE_AA)

cv2.putText(self.im_rd, "Warning: please shortcut having rectangle", (50, 140), font, 0.8, (0, 0, 255), 1,

cv2.LINE_AA)

# print(im_rd.shape)

height, width = self.im_rd.shape[:2]

image1 = cv2.cvtColor(self.im_rd, cv2.COLOR_BGR2RGB)

pic = wx.Bitmap.FromBuffer(width, height, image1)

# 顯示圖片在panel上

self.bmp.SetBitmap(pic)

#直接在sbclicked里設置self.bmp.SetBitmap(wx.Bitmap(self.image_cover))，子線程還在運行

if self.NewButton.IsEnabLED()==True and self.ShortCutButton.IsEnabled()==False and self.SaveButton.IsEnabled()==False:

self.bmp.SetBitmap(wx.Bitmap(self.image_info))

_thread.exit()

總結

以上是生活随笔為你收集整理的python参数检验框架_基于python的人员识别检测框架研究的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。