2
0
mirror of https://github.com/Shawn-Shan/fawkes.git synced 2024-12-22 07:09:33 +05:30

small fixes

This commit is contained in:
Shawn-Shan 2021-04-21 10:55:15 -05:00
parent b35f7eb0ab
commit 8ff3175cea
2 changed files with 10 additions and 15 deletions

View File

@ -46,13 +46,9 @@ class Fawkes(object):
self.th = th self.th = th
self.lr = lr self.lr = lr
self.max_step = max_step self.max_step = max_step
if gpu is not None:
init_gpu(gpu) init_gpu(gpu)
# model_dir = os.path.join(os.path.expanduser('~'), '.fawkes')
# if not os.path.exists(os.path.join(model_dir)):
# os.makedirs(model_dir, exist_ok=True)
self.aligner = aligner() self.aligner = aligner()
self.protector = None self.protector = None
@ -61,20 +57,20 @@ class Fawkes(object):
def mode2param(self, mode): def mode2param(self, mode):
if mode == 'low': if mode == 'low':
th = 0.005 th = 0.0045
max_step = 25 max_step = 25
lr = 25 lr = 25
extractors = ["extractor_2"] extractors = ["extractor_2"]
elif mode == 'mid': elif mode == 'mid':
th = 0.01 th = 0.012
max_step = 50 max_step = 50
lr = 20 lr = 20
extractors = ["extractor_0", "extractor_2"] extractors = ["extractor_0", "extractor_2"]
elif mode == 'high': elif mode == 'high':
th = 0.015 th = 0.017
max_step = 100 max_step = 150
lr = 15 lr = 15
extractors = ["extractor_0", "extractor_2"] extractors = ["extractor_0", "extractor_2"]
@ -82,7 +78,7 @@ class Fawkes(object):
raise Exception("mode must be one of 'min', 'low', 'mid', 'high'") raise Exception("mode must be one of 'min', 'low', 'mid', 'high'")
return th, max_step, lr, extractors return th, max_step, lr, extractors
def run_protection(self, image_paths, th=0.04, sd=1e9, lr=10, max_step=500, batch_size=1, format='png', def run_protection(self, image_paths, th=0.04, sd=1e7, lr=10, max_step=500, batch_size=1, format='png',
separate_target=True, debug=False, no_align=False, exp="", maximize=True, separate_target=True, debug=False, no_align=False, exp="", maximize=True,
save_last_on_failed=True): save_last_on_failed=True):
@ -117,7 +113,7 @@ class Fawkes(object):
learning_rate=self.lr, learning_rate=self.lr,
max_iterations=self.max_step, max_iterations=self.max_step,
l_threshold=self.th, l_threshold=self.th,
verbose=0, verbose=debug,
maximize=maximize, maximize=maximize,
keep_final=False, keep_final=False,
image_shape=(IMG_SIZE, IMG_SIZE, 3), image_shape=(IMG_SIZE, IMG_SIZE, 3),
@ -137,7 +133,7 @@ class Fawkes(object):
continue continue
p_img = final_images[i] p_img = final_images[i]
path = image_paths[i] path = image_paths[i]
file_name = "{}_cloaked.{}".format(".".join(path.split(".")[:-1]), format) file_name = "{}_{}_cloaked.{}".format(".".join(path.split(".")[:-1]), self.mode, format)
dump_image(p_img, file_name, format=format) dump_image(p_img, file_name, format=format)
print("Done!") print("Done!")

View File

@ -190,9 +190,8 @@ class Faces(object):
if not self.no_align: if not self.no_align:
cur_index = align_img[1] cur_index = align_img[1]
self.cropped_faces_shape.extend(cur_shapes) self.cropped_faces_shape.extend(cur_shapes)
self.cropped_index.extend(cur_index) self.cropped_index.extend(cur_index[:len(cur_faces_square)])
self.callback_idx.extend([i] * len(cur_faces_square)) self.callback_idx.extend([i] * len(cur_faces_square))
if len(self.cropped_faces) == 0: if len(self.cropped_faces) == 0: