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-24 14:07:25 -04:00
parent f352bfd25d
commit 9c2cb063be
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stargan/solver.py
+4 -2
View File
@@ -610,8 +610,10 @@ class Solver(object):
# x_adv, perturb = pgd_attack.perturb(x_real, x_real, c_trg_list[0])
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, perturb = pgd_attack.perturb(x_real, gen_noattack, c_trg)
# x_adv = x_real + perturb
# x_adv = self.blur_tensor(x_adv)
@@ -627,7 +629,7 @@ class Solver(object):
# No Attack
# gen_noattack, _ = self.G(x_real, c_trg)
gen_noattack, gen_noattack_feats = self.G(x_real, c_trg)
# gen_noattack, gen_noattack_feats = self.G(x_real, c_trg)
l1_error += F.l1_loss(gen, gen_noattack)
l2_error += F.mse_loss(gen, gen_noattack)