From f2ad8b7b4788bcd7aaa3c93515ebeee7cca08d0b Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 11:38:39 -0400 Subject: [PATCH] next --- pix2pixHD_attack/test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pix2pixHD_attack/test.py b/pix2pixHD_attack/test.py index f6f90d9..f232bbc 100755 --- a/pix2pixHD_attack/test.py +++ b/pix2pixHD_attack/test.py @@ -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)),