CalvinBackup/disrupting-deepfakes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

next

Browse Source
This commit is contained in:
Nataniel Ruiz
2019-12-26 10:05:58 -04:00
parent 67740119d9
commit 154b42d425
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stargan/solver.py
+7 -3
View File
@@ -609,15 +609,19 @@ class Solver(object):
# Translate images.
x_fake_list = [x_real]
x_advs = {}
if i == 0:
x_adv, perturb = pgd_attack.perturb(x_real, x_real, c_trg_list[0])
for c_trg in c_trg_list:
x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg)
x_advs[c_trg] = x_adv, perturb
for c_trg in c_trg_list:
with torch.no_grad():
gen_noattack, gen_noattack_feats = self.G(x_real, c_trg)
# Attack
x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg)
# x_adv = x_real + perturb
# x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg)
_, perturb = x_advs[c_trg]
x_adv = x_real + perturb
# x_adv = self.blur_tensor(x_adv)
# Metrics