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

app

Browse Source 
Former-commit-id: 0282ba3b18eab09351540df5aa9858d466aa3454 [formerly 88ce02e785b35491c0fe9ffec50d9504c6069780]
Former-commit-id: d1d8b474c1383b05dca7e3e6084847b6e852c3ad
This commit is contained in:
Shawn-Shan 2020-07-08 23:10:13 -05:00
parent 38f89999fa
commit 448c58ab16
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
fawkes/protection.py
View File

@ -70,7 +70,7 @@ def main(*argv):
help='GPU id', default='0') help='GPU id', default='0')
parser.add_argument('--mode', '-m', type=str, parser.add_argument('--mode', '-m', type=str,
help='cloak generation mode', default='mid') help='cloak generation mode', default='low')
parser.add_argument('--feature-extractor', type=str, parser.add_argument('--feature-extractor', type=str,
help="name of the feature extractor used for optimization", help="name of the feature extractor used for optimization",
default="high_extract") default="high_extract")
@ -95,21 +95,21 @@ def main(*argv):
args.lr = 20 args.lr = 20
elif args.mode == 'mid': elif args.mode == 'mid':
args.feature_extractor = "high_extract" args.feature_extractor = "high_extract"
args.th = 0.004 args.th = 0.005
args.max_step = 50 args.max_step = 50
args.lr = 15 args.lr = 15
elif args.mode == 'high': elif args.mode == 'high':
args.feature_extractor = "high_extract" args.feature_extractor = "high_extract"
args.th = 0.007 args.th = 0.008
args.max_step = 100 args.max_step = 500
args.lr = 10 args.lr = 15
elif args.mode == 'ultra': elif args.mode == 'ultra':
if not tf.test.is_gpu_available(): if not tf.test.is_gpu_available():
print("Please enable GPU for ultra setting...") print("Please enable GPU for ultra setting...")
sys.exit(1) sys.exit(1)
# args.feature_extractor = ["high_extract", 'high2_extract'] # args.feature_extractor = ["high_extract", 'high2_extract']
args.feature_extractor = "high_extract" args.feature_extractor = "high_extract"
args.th = 0.015 args.th = 0.01
args.max_step = 2000 args.max_step = 2000
args.lr = 8 args.lr = 8
elif args.mode == 'custom': elif args.mode == 'custom':