“““
數(shù)據處理相關功能
“““
import?os
import?cv2
import?numpy?as?np
from?sklearn.model_selection?import?train_test_split


def?read_img（path）:
????“““
????功能：讀取圖片
????#?參數(shù)：
????????path:?數(shù)據集的路徑
????#?返回：
????????res:?不同人的圖片。
????“““
????res?=?[]

????for?（root?dirs?files）?in?os.walk（path）:
????????if?files:
????????????tmp?=?[]
????????????files?=?np.random.choice（files?4）
????????????for?f?in?files:
????????????????img?=?os.path.join（root?f）
????????????????image?=?cv2.imread（img）
????????????????image?=?cv2.resize（image?（64?64）
???????????????????????????????????interpolation=cv2.INTER_CUBIC）
????????????????image?=?np.array（image?dtype=‘float32‘）
????????????????image?/=?255.
????????????????tmp.append（image）

????????????res.append（tmp）

????return?res


def?get_paris（path）:
????“““
????功能：構造用于訓練的成對數(shù)據
????#?參數(shù)：
????????path:?數(shù)據集的路徑
????#?返回：
????????sm1:?一對數(shù)據中的第一個對象
????????sm2:?一對數(shù)據中的第二個對象
????????y1:?成對數(shù)據的標簽，相同為1，不同為0.
????“““
????sm1?sm2?df1?df2?=?[]?[]?[]?[]
????res?=?read_img（path）

????persons?=?len（res）

????for?i?in?range（persons）:
????????for?j?in?range（i?persons）:
????????????p1?=?res[i]
????????????p2?=?res[j]

????????????if?i?==?j:
????????????????for?pi?in?p1:
????????????????????for?pj?in?p2:
????????????????????????sm1.append（pi）
????????????????????????sm2.append（pj）
????????????else:
????????????????df1.append（p1[0]）
????????????????df2.append（p2[0]）

????df1?=?df1[:len（sm1）]
????df2?=?df2[:len（sm2）]
????y1?=?list（np.zeros（len（sm1）））
????y2?=?list（np.ones（len（df1）））

????sm1.extend（df1）
????sm2.extend（df2）
????y1.extend（y2）

????return?sm1?sm2?y1


def?create_generator（x?y?batch）:
????“““
????功能：構造數(shù)據生成器
????#?參數(shù)：
????????x:?數(shù)據
????????y:?標簽
????????batch:?數(shù)據生成器每次生成數(shù)據的個數(shù)
????#?返回：
????????[x1?x2]:?成對數(shù)據
????????yb:?數(shù)據標簽
????“““
????while?True:
????????index?=?np.random.choice（len（y）?batch）?#?每次選擇batch個數(shù)據索引
????????x1?x2?yb?=?[]?[]?[]
????????for?i?in?index:
????????????x1.append（x[i][0]）
????????????x2.append（x[i][1]）
????????????yb.append（y[i]）
????????x1?=?np.array（x1）
????????x2?=?np.array（x2）

????????yield?[x1?x2]?yb


def?get_train_test（path）:
????“““
????功能：切分數(shù)據集
????#?參數(shù)：
????????path:?數(shù)據集的路徑
????#?返回：
????????X_train:?用于訓練的數(shù)據
????????X_test:?用于測試的數(shù)據
????????y_train:?用于訓練的標簽
????????y_test:?用于測試的標簽
????“““
????im1?im2?y?=?get_paris（path）?#?構造成對數(shù)據
????im?=?list（zip（im1?im2））

????X_train?X_test?y_train?y_test?=?train_test_split（
????????im?y?test_size=0.33）??#?利用sklearn切分數(shù)據集
????return?X_train?X_test?y_train?y_test

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----

?????文件????????135??2019-02-28?16:51??FaceRecognitionUncompleted\.idea\encodings.xml

?????文件????????528??2019-12-19?22:27??FaceRecognitionUncompleted\.idea\FaceRecognitionUncompleted.iml

?????文件???????2085??2019-12-19?09:33??FaceRecognitionUncompleted\.idea\inspectionProfiles\Project_Default.xml

?????文件????????214??2019-12-19?22:27??FaceRecognitionUncompleted\.idea\misc.xml

?????文件????????304??2019-02-28?16:51??FaceRecognitionUncompleted\.idea\modules.xml

?????文件??????34650??2019-12-21?14:46??FaceRecognitionUncompleted\.idea\workspace.xml

?????文件???????3587??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.1.jpg

?????文件???????3562??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.10.jpg

?????文件???????3573??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.11.jpg

?????文件???????3635??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.12.jpg

?????文件???????3591??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.13.jpg

?????文件???????3625??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.14.jpg

?????文件???????3653??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.15.jpg

?????文件???????3654??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.16.jpg

?????文件???????3673??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.17.jpg

?????文件???????3689??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.18.jpg

?????文件???????3667??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.19.jpg

?????文件???????3553??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.2.jpg

?????文件???????3665??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.20.jpg

?????文件???????3562??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.3.jpg

?????文件???????3621??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.4.jpg

?????文件???????3648??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.5.jpg

?????文件???????3628??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.6.jpg

?????文件???????3622??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.7.jpg

?????文件???????3617??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.8.jpg

?????文件???????3584??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\and\and_exp.9.jpg

?????文件???????3881??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\ant\ant_exp.1.jpg

?????文件???????3876??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\ant\ant_exp.10.jpg

?????文件???????3838??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\ant\ant_exp.11.jpg

?????文件???????3894??2019-02-28?16:51??FaceRecognitionUncompleted\data\dataset\ant\ant_exp.12.jpg

............此處省略396個文件信息