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-30 14:46:13 -04:00
parent 75c179aa1e
commit 26b96d9ab7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pix2pixHD_attack/test.py
+2 -1
View File
@@ -69,7 +69,8 @@ for i, data in enumerate(dataset):
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)),
visuals = OrderedDict([('original_label', util.tensor2label(data['label'], opt.label_nc)),
('input_label', util.tensor2label(adv_img.data[0], opt.label_nc)),
('attacked_image', util.tensor2im(generated.data[0])),
('noattack', util.tensor2im(generated_noattack.data[0]))])
img_path = data['path']