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-26 11:38:39 -04:00
parent ebebe20778
commit f2ad8b7b47
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pix2pixHD_attack/test.py
+3 -3
View File
@@ -57,15 +57,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_noattack = model.inference(data['label'], data['inst'], data['image'])
adv_image, perturb = model.attack(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)),