From 98f65c02e9027d8976d5fa7666647edb92d90ab3 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Fri, 10 Jan 2020 12:53:09 -0400 Subject: [PATCH] GANimation conditional attacks --- ganimation/solver.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ganimation/solver.py b/ganimation/solver.py index e31cbea..8dc9e43 100644 --- a/ganimation/solver.py +++ b/ganimation/solver.py @@ -416,14 +416,14 @@ 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 # x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets[0, :].unsqueeze(0).cuda()) # Joint Class Conditional - x_adv, perturb = pgd_attack.perturb_joint_class(image_to_animate, black, targets[:, :].cuda()) + # 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()) @@ -438,10 +438,10 @@ class Solver(Utils): # 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 + x_adv = image_to_animate # print(image_to_animate.shape, x_adv.shape)