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-25 15:23:45 -04:00
parent 16857d1569
commit 79fa680c1b
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pix2pixHD_attack/models/pix2pixHD_model.py
View File

@@ -241,7 +241,7 @@ class Pix2PixHDModel(BaseModel):
else:
input_concat = input_label
input_adv = torch.clamp(input_concat + perturb * 1000, min=-1, max=1)
input_adv = torch.clamp(input_concat + perturb, min=-1, max=1)
with torch.no_grad():
fake_image = self.netG.forward(input_adv)

5
pix2pixHD_attack/test.py
View File

@@ -52,14 +52,15 @@ for i, data in enumerate(dataset):
exit(0)
minibatch = 1
if i == 0:
adv_image, perturb = model.attack(data['label'], data['inst'], data['image'])
# if i == 0:
# adv_image, perturb = model.attack(data['label'], data['inst'], data['image'])
if opt.engine:
generated = run_trt_engine(opt.engine, minibatch, [data['label'], data['inst']])
elif opt.onnx:
generated = run_onnx(opt.onnx, opt.data_type, minibatch, [data['label'], data['inst']])
else:
# generated = model.inference(data['label'], data['inst'], data['image'])
adv_image, perturb = model.attack(data['label'], data['inst'], data['image'])
generated, adv_img = model.inference_attack(data['label'], data['inst'], data['image'], perturb)
visuals = OrderedDict([('input_label', util.tensor2label(adv_img.data[0], opt.label_nc)),

2
pix2pixHD_attack/util/attacks.py
View File

@@ -7,7 +7,7 @@ import torch
import torch.nn as nn
class LinfPGDAttack(object):
def __init__(self, model=None, epsilon=0.2, k=1, a=0.01):
def __init__(self, model=None, epsilon=0.05, k=1, a=0.05):
self.model = model
self.epsilon = epsilon
self.k = k