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 15:44:50 -04:00
parent 9f4abd81b2
commit 07f7da1603
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
@@ -58,15 +58,15 @@ for i, data in enumerate(dataset):
minibatch = 1
# Transfer
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([('original_label', util.tensor2label(data['label'][0], opt.label_nc)),