1. 程式人生 > >facenet 人臉識別(一)

facenet 人臉識別(一)

前言

已完成TensorFlow Object Detection API環境搭建,具體搭建過程請參照:

下載Facenet原始碼工程

2. 將下載的原始碼解壓,如圖所示:

 

安裝和配置Facenet環境

1. 在自己電腦對應的Anaconda3\Lib\site-packages目錄下,新建facenet資料夾,本人的目錄如下:

2. 然後,將facenet-master\src目錄下的全部檔案複製到上面新建的facenet資料夾內;

facenet-master\src目錄下的全部檔案資訊如下:

複製到facenet目錄內,如下:

3. 最後,在Anaconda Prompt內輸入import facenet,不會報錯即可,如下:

下載LFW資料集


1. LFW資料集是由美國馬薩諸塞大學阿姆斯特分校計算機視覺實驗室整理的人臉檢測資料集,是評估人臉識別演算法效果的公開測試資料集,全稱為帶標籤的自然人臉資料庫(Labeled Faces in the Wild);

2. LFW資料庫內每張圖片命名方式為“lfw/name/name_xxxx.jpg”,這裡“xxxx”是前面補零的四點陣圖片編號。例如,前美國總統喬治•W•布什的第10張圖片為“lfw/George_W_Bush/George_W_Bush_0010.jpg”。

3. LFW資料庫 總共有 13233 張 JPEG 格式圖片,屬於 5749 個不同人。每張圖片尺寸都是 250x250;

5. 下載完成後,把資料集解壓到facenet-master\data下面,如下:

 

對LFW資料集進行預處理


1. 在data目錄新建一個空資料夾,命名為“lfw_160”;

原影象大小為250*250,如下圖:

2. 我們需要將待檢測所使用的資料集校準為和預訓練模型所使用的資料集大小一致(160*160),轉換後的資料集儲存在lfw_160資料夾內;

3.切換目錄至facenet-master下

4.圖片預處理——執行人臉對齊程式(src\align\align_dataset_mtcnn.py)

命令校準

python src\align\align_dataset_mtcnn.py --help

 

缺少align模組,進行安裝

pip install align -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

再次執行

python src\align\align_dataset_mtcnn.py --help

需配置環境變數

再輸入

python src\align\align_dataset_mtcnn.py --help

 執行轉換

python src\align\align_dataset_mtcnn.py data/lfw data/lfw_160 --image_size 160 --margin 32 --random_order --gpu_memory_fraction 0.25

下載訓練好的網路模型


1. facenet提供了兩個預訓練模型,分別是基於CASIA-WebFace和MS-Celeb-1M人臉庫訓練的,如下:


2. 本人使用的是基於資料集CASIA-WebFace採用Inception ResNet v1神經網路結構訓練好的模型。模型儲存在Google網盤,需要翻牆。推薦大家用Lantern;

3. 把下載的檔案解壓到src\models\目錄下面,如下所示:

4. 程式也下載了,測試資料集LFW也有了,模型也有了,接下來就是評估模型的準確率;

評估預訓練模型的準確率


1. 在cmd命令列或者Anaconda Propmt下定位到facenet資料夾下;

cd D:\eclipse-workspace\facenet-master

2. 輸入以下命令:

python src\validate_on_lfw.py D:\eclipse-workspace\facenet-master\data\lfw_160 D:\eclipse-workspace\facenet-master\models\20180408-102900

預測中,結果如圖:

人臉對比

1. Facenet可以直接對比2個人臉經過它的網路對映之後的歐式距離;

執行程式為facenet-master\src\compare.py;

2.在data\images目錄下有兩張圖片進行測試;

python src\compare.py D:\eclipse-workspace\facenet-master\models\20180408-102900 D:\eclipse-workspace\facenet-master\data\images\Anthony_Hopkins_0001.jpg D:\eclipse-workspace\facenet-master\data\images\Anthony_Hopkins_0002.jpg