支持白盒攻击facenet

Joejiong · Sep 17, 2018 · 9b5d6f3 · 9b5d6f3
1 parent 361c9ee
commit 9b5d6f3
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,7 +3,9 @@
 *.so
 *.egg
 *.egg-info
+*.pb
 dist
 build
 
 .idea/
+thirdparty/
diff --git a/applications/face_recognition_attack/Bill_Gates_0001_2_Michael_Jordan_00021114.png b/applications/face_recognition_attack/Bill_Gates_0001_2_Michael_Jordan_00021114.png
diff --git a/applications/face_recognition_attack/README.md b/applications/face_recognition_attack/README.md
@@ -1,6 +1,42 @@
 # Face Recogniztion Attack
 
-## white box attack
+## White box attack
 We chose a pre-trained FaceNet model that is a state-of-the-art and widely used face-recognition system as our white-box attacked model. We used gradient-based attacks methods and modify its loss function using FaceNet embedding distance.
 
-FaceNet and its pre-trained model can be downloaded from `https://github.com/davidsandberg/facenet`.
+## Install
+
+FaceNet and its pre-trained model can be downloaded from `https://github.com/davidsandberg/facenet`.
+
+	cd applications/
+
+### Clone facenet
+
+	git clone https://github.com/davidsandberg/facenet.git
+
+### Download pre-trained model
+
+baidu netdisk https://pan.baidu.com/s/1gF6iwpF77UUHkibQY6QwMA
+
+you can get 20180402-114759.zip，unzip it，get 20180402-114759.pb
+
+	cd applications/face_recognition_attack
+	total 93600
+	drwxr-xr-x 3 root root     4096 Sep 17 16:59 ./
+	drwxr-xr-x 3 root root     4096 Sep 17 16:33 ../
+	-rw-r--r-- 1 root root 95745767 Apr  9 14:45 20180402-114759.pb
+	-rw-r--r-- 1 root root    34639 Sep 17 16:33 Bill_Gates_0001.png
+	drwxr-xr-x 9 root root     4096 Sep 17 16:58 facenet/
+	-rw-r--r-- 1 root root     6114 Sep 17 16:33 facenet_fr.py
+	-rw-r--r-- 1 root root    36618 Sep 17 16:33 Michael_Jordan_0002.png
+	-rw-r--r-- 1 root root      388 Sep 17 16:33 README.md
+
+## Attack face recogniztion
+
+
+
+![Bill_Gates_0001.png](Bill_Gates_0001.png)
+![Michael_Jordan_0002.png](Michael_Jordan_0002.png)
+
+	python facenet_fr.py
+
+![Bill_Gates_0001_2_Michael_Jordan_00021114.png](Bill_Gates_0001_2_Michael_Jordan_00021114.png)
diff --git a/applications/face_recognition_attack/facenet_fr.py b/applications/face_recognition_attack/facenet_fr.py
@@ -22,10 +22,10 @@
 import os
 import time
 
-sys.path.append("../../../facenet/src")
+sys.path.append("../thirdparty/facenet/src")
 import facenet
 
-FACENET_MODEL_CHECKPOINT = "/mnt/model/facenet/20180402-114759/20180402-114759.pb"
+FACENET_MODEL_CHECKPOINT = "20180402-114759.pb"
 
 
 def get_pic_from_png(pic_path):