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

GANimation conditional attacks

Browse Source
This commit is contained in:
Nataniel Ruiz
2020-01-09 17:57:43 -04:00
parent 64a5363832
commit bbaa697c36
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ganimation/attacks.py
View File

@@ -96,7 +96,7 @@ class LinfPGDAttack(object):
self.model.zero_grad()
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
@@ -131,7 +131,7 @@ class LinfPGDAttack(object):
out = imFromAttReg(output_att, output_img, X)
loss = -self.loss_fn(output_att, y)
loss = self.loss_fn(output_att, y)
# loss = -self.loss_fn(out, y)
full_loss += loss

8
ganimation/solver.py
View File

@@ -416,7 +416,7 @@ class Solver(Utils):
for target_idx in range(targets.size(0)):
# Transfer to different classes
# if target_idx == 0:
if target_idx == 0:
# img = regular_image_transform(Image.open(images_to_animate_path[idx])).unsqueeze(0).cuda()
# Wrong Class
@@ -426,7 +426,7 @@ class Solver(Utils):
# x_adv, perturb = pgd_attack.perturb_joint_class(image_to_animate, black, targets[:, :].cuda())
# Iterative Class Conditional
# x_adv, perturb = pgd_attack.perturb_iter_class(image_to_animate, black, targets[:, :].cuda())
x_adv, perturb = pgd_attack.perturb_iter_class(image_to_animate, black, targets[:, :].cuda())
# Iterative Data
# _, perturb = pgd_attack.perturb_iter_data(image_to_animate, all_images, black, targets[68, :].unsqueeze(0).cuda())
@@ -434,11 +434,11 @@ class Solver(Utils):
targets_au = targets[target_idx, :].unsqueeze(0).cuda()
# Normal Attack
x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets_au)
# x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets_au)
# x_adv, perturb = x_advs[target_idx]
# x_adv = image_to_animate + perturb
x_adv = image_to_animate + perturb
# No Attack
# x_adv = image_to_animate