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-31 21:12:17 -04:00
parent 5dcfd711f3
commit 97658b02c8
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ganimation/solver.py
+7 -7
View File
@@ -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(