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 11:57:32 -04:00
parent 3859120822
commit 5caf178b62
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ganimation/attacks.py
+1 -2
View File
@@ -31,8 +31,7 @@ class LinfPGDAttack(object):
self.model.zero_grad()
# loss = -self.loss_fn(output_att, y) + self.loss_fn(output_img, y)
# loss = self.loss_fn(output_att, y)
loss = self.loss_fn(output_att, y)
loss = -self.loss_fn(out, y)
loss.backward()
grad = X.grad
ganimation/solver.py
+4 -4
View File
@@ -436,10 +436,10 @@ class Solver(Utils):
image_path.split('/')[-1].split('.')[0]
+ '_ref.jpg'))
l1_error += F.l1_loss(resulting_image, gen_noattack)
l2_error += F.mse_loss(resulting_image, gen_noattack)
l0_error += (resulting_image - gen_noattack).norm(0)
min_dist += (resulting_image - gen_noattack).norm(float('-inf'))
l1_error += F.l1_loss(resulting_image, resulting_image_noattack)
l2_error += F.mse_loss(resulting_image, resulting_image_noattack)
l0_error += (resulting_image - resulting_image_noattack).norm(0)
min_dist += (resulting_image - resulting_image_noattack).norm(float('-inf'))
n_samples += 1
# Print metrics