From 26b96d9ab7472bafc13435ad93387a95915aee02 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Mon, 30 Dec 2019 14:46:13 -0400 Subject: [PATCH] next --- pix2pixHD_attack/test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pix2pixHD_attack/test.py b/pix2pixHD_attack/test.py index 7a0ddb3..290b1f8 100755 --- a/pix2pixHD_attack/test.py +++ b/pix2pixHD_attack/test.py @@ -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']