From aeb56eef5635c4a2080be5f38e2f701f0a642c39 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=90=9B=E5=90=9B=E8=87=A3=E8=87=A3=E5=90=9B?= Date: Fri, 30 Sep 2022 05:45:46 +0000 Subject: [PATCH] update examples/community/face_adversarial_attack/example_non-target_attack.py. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 君君臣臣君 --- .../face_adversarial_attack/example_non-target_attack.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/community/face_adversarial_attack/example_non-target_attack.py b/examples/community/face_adversarial_attack/example_non-target_attack.py index 4861c4b..df4c6f0 100644 --- a/examples/community/face_adversarial_attack/example_non-target_attack.py +++ b/examples/community/face_adversarial_attack/example_non-target_attack.py @@ -16,7 +16,7 @@ import numpy as np import matplotlib.image as mp from mindspore import context -import ad +import adversarial_attack context.set_context(mode=context.GRAPH_MODE, device_target="GPU") @@ -25,11 +25,11 @@ context.set_context(mode=context.GRAPH_MODE, device_target="GPU") if __name__ == '__main__': - inputs = AFR.load_data('photos/input/') - targets = AFR.load_data('photos/target/') + inputs = adversarial_attack.load_data('photos/input/') + targets = adversarial_attack.load_data('photos/target/') - adversarial = AFR.FaceAdversarialAttack(inputs[0], targets[0]) + adversarial = adversarial_attack.FaceAdversarialAttack(inputs[0], targets[0]) attack_method = "non-target attack" adversarial_tensor, mask_tensor = adversarial.train(attack_method)