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

bug fixes

Browse Source
This commit is contained in:
Shawn-Shan 2020-07-24 11:11:42 -05:00
parent 3775bba1b4
commit 1e8893cacb
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
fawkes/differentiator.py
View File

@ -407,7 +407,6 @@ class FawkesMaskGeneration:
if iteration != 0 and iteration % (self.MAX_ITERATIONS // 3) == 0:
LR = LR * 0.8
print("LR: {}".format(LR))
if iteration % (self.MAX_ITERATIONS // 5) == 0:
if self.verbose == 1:

6
fawkes/protection.py
View File

@ -81,8 +81,8 @@ class Fawkes(object):
print("Please enable GPU for ultra setting...")
sys.exit(1)
th = 0.01
max_step = 200
lr = 20
max_step = 1000
lr = 10
else:
raise Exception("mode must be one of 'low', 'mid', 'high', 'ultra', 'custom'")
return th, max_step, lr
@ -179,7 +179,7 @@ def main(*argv):
default="high_extract")
parser.add_argument('--th', type=float, default=0.01)
parser.add_argument('--max-step', type=int, default=500)
parser.add_argument('--max-step', type=int, default=1000)
parser.add_argument('--sd', type=int, default=1e9)
parser.add_argument('--lr', type=float, default=2)

2
setup.py
View File

@ -77,7 +77,7 @@ setup_requires = []
install_requires = [
'numpy==1.16.4',
# 'tensorflow-gpu>=1.13.1, <=1.14.0',
'tensorflow>=1.12.0, <=1.15.0', # change this is tensorflow-gpu if using GPU machine.
'tensorflow>=1.12.0, <=1.15.0', # change this is tensorflow-gpu if using GPU machine.
'argparse',
'keras>=2.2.5, <=2.3.1',
'scikit-image',