#coding=utf-8
import?os
import?cv2

import?numpy

def?mkdir（filename）:
????‘‘‘
????創(chuàng)建文件夾
????:param?filename:?文件夾目錄
????:return:?空
????‘‘‘
????‘‘‘
????判斷是否存在該文件夾
????‘‘‘
????if?not?os.path.exists（filename）:
????????‘‘‘
????????創(chuàng)建文件夾
????????‘‘‘
????????os.mkdir（filename）

def?readFace（）:
????‘‘‘
????裁剪人臉照片
????:return:
????‘‘‘
????fileFacebase?=?“face“
????‘‘‘
????創(chuàng)建人臉根目錄
????‘‘‘
????mkdir（fileFacebase）
????‘‘‘
????獲取數(shù)據(jù)目錄子級(jí)列表
????‘‘‘
????dirs?=?os.listdir（“d_img“）
????‘‘‘
????循環(huán)獲取目錄信息
????‘‘‘
????for?j?dir?in?enumerate（dirs）:
????????‘‘‘
????????拼接人臉子集目錄路徑
????????‘‘‘
????????fileInfo?=?fileFacebase?+?“/%s“?%?str（dir）
????????‘‘‘
????????創(chuàng)建人臉自己目錄
????????‘‘‘
????????mkdir（fileInfo）
????????‘‘‘
????????獲取采集數(shù)據(jù)子集的子集文件
????????‘‘‘
????????ds?=?os.listdir（“d_img/%s“?%?dir）
????????‘‘‘
????????定義變量方便人臉文件命名
????????‘‘‘
????????num?=?0
????????‘‘‘
????????遍歷采集數(shù)據(jù)子集的子集文件
????????‘‘‘
????????for?d?in?ds:
????????????‘‘‘
????????????拼接文件路徑
????????????‘‘‘
????????????path?=?“d_img/%s/%s“?%?（dir?d）
????????????‘‘‘
????????????讀取文件
????????????‘‘‘
????????????image?=?cv2.imread（path）
????????????try:
????????????????‘‘‘
????????????????更改圖片為灰度圖片加快辨認(rèn)速度（可選但是要統(tǒng)一）
????????????????‘‘‘
????????????????image?=?cv2.cvtColor（src=image?code=cv2.COLOR_BGR2GRAY）
????????????except?Exception?as?e:
????????????????pass

????????????‘‘‘
????????????加載臉部數(shù)據(jù)對(duì)比
????????????‘‘‘
????????????face_detecotr?=?cv2.CascadeClassifier（“haarcascade_frontalface_alt.xml“）
????????????‘‘‘
????????????對(duì)比數(shù)據(jù)
????????????‘‘‘
????????????face_dete?=?face_detecotr.detectMultiScale（image?minNeighbors=4）
????????????‘‘‘
????????????判斷是否可以采集到人臉數(shù)據(jù)
????????????‘‘‘
????????????if?isinstance（face_dete?tuple）:
????????????????print（“沒(méi)有檢查到頭像111“）
????????????????pass
????????????else:
????????????????‘‘‘
????????????????采集到數(shù)據(jù)名稱(chēng)加1
????????????????‘‘‘
????????????????num?+=?1
????????????????‘‘‘
????????????????遍歷數(shù)據(jù)獲取起點(diǎn)xy軸信息人臉寬度w高度h
????????????????‘‘‘
????????????????for?x?y?w?h?in?face_dete:
????????????????????‘‘‘
????????????????????繪畫(huà)矩形框便于展示
????????????????????1窗口展示的圖片
????????????????????2起點(diǎn)xy坐標(biāo)
????????????????????3終點(diǎn)xy坐標(biāo)
????????????????????4矩形框顏色
????????????????????5邊框?qū)挾?br/>????????????????????‘‘‘
????????????????????cv2.rectangle（img=image?pt1=（x?y）?pt2=（x?+?w?y?+?h）?color=[0?0?255]?thickness=1）
????????????????????‘‘‘
????????????????????裁剪獲取人臉部分?jǐn)?shù)據(jù)
????????????????????‘‘‘
????????????????????imgFace?=?image[y:y?+?h?x:x?+?w]
????????????????????‘‘‘
????????????????????設(shè)置圖片大小
????????????????????‘‘‘
????????????????????imgFace?=?cv2.resize（imgFace?dsize=（200?200））
????????????????????‘‘‘
????????????????????保存人臉圖片
????????????????????‘‘‘
????????????????????cv2.imwrite（“face/%s/%d_face.jpg“?%?（dir?num）?imgFace）
‘‘‘
獲取人臉數(shù)據(jù)
‘‘‘
def?getFace（）:
????‘‘‘
????獲取人臉文件夾下所有子文件夾
????‘‘‘
????dirs?=?os.listdir（“face“）
????‘‘‘
????因?yàn)樾枰屗惴▽W(xué)習(xí)所以將數(shù)據(jù)遍歷之?dāng)?shù)組中方便算法學(xué)習(xí)
????‘‘‘
????#?用于存放人臉數(shù)據(jù)
????X?=?[]
????#?用于存放對(duì)應(yīng)數(shù)據(jù)的標(biāo)簽猜想應(yīng)該可以直接使用文件夾名稱(chēng)
????Y?=?[]
????‘‘‘
????遍歷人臉圖片文件存儲(chǔ)路徑將輸入存儲(chǔ)到數(shù)據(jù)集合
????‘‘‘

?屬性????????????大小?????日期????時(shí)間???名稱(chēng)
-----------?---------??----------?-----??----
?????文件????????8553??2019-10-24?20:33??python人臉識(shí)別\05_opencv人臉識(shí)別.py
?????文件??????341406??2019-10-23?09:36??python人臉識(shí)別\haarcascade_eye.xml
?????文件??????676709??2019-10-23?09:36??python人臉識(shí)別\haarcascade_frontalface_alt.xml
?????文件?????????242??2020-03-13?00:18??python人臉識(shí)別\requirements.txt
?????目錄???????????0??2020-03-13?00:20??python人臉識(shí)別\