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)