From 97658b02c8b0ea6a34c670f8da346c3d03ecccf0 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 31 Dec 2019 21:12:17 -0400 Subject: [PATCH] next --- ganimation/solver.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/ganimation/solver.py b/ganimation/solver.py index 1e9ac73..e52f633 100644 --- a/ganimation/solver.py +++ b/ganimation/solver.py @@ -409,10 +409,10 @@ class Solver(Utils): all_images = torch.cat([regular_image_transform(Image.open(path)).unsqueeze(0) for path in images_to_animate_path], dim=0).cuda() - if idx == 0: - for target_idx in range(targets.size(0)): - x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets[target_idx, :].unsqueeze(0).cuda()) - x_advs.append((x_adv, perturb)) + # if idx == 0: + # for target_idx in range(targets.size(0)): + # x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets[target_idx, :].unsqueeze(0).cuda()) + # x_advs.append((x_adv, perturb)) for target_idx in range(targets.size(0)): # if target_idx == 0: @@ -422,9 +422,9 @@ class Solver(Utils): # # _, perturb = pgd_attack.perturb_iter_data(image_to_animate, all_images, black, targets[68, :].unsqueeze(0).cuda()) targets_au = targets[target_idx, :].unsqueeze(0).cuda() - # 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, perturb = pgd_attack.perturb(image_to_animate, black, targets_au) + # x_adv, perturb = x_advs[target_idx] + # x_adv = image_to_animate + perturb # print(image_to_animate.shape, x_adv.shape) with torch.no_grad(): resulting_images_att, resulting_images_reg = self.G(