diff --git a/examples/face_adversarial_attack/example/example_non-target_attack.py b/examples/face_adversarial_attack/example/example_non-target_attack.py index a1fd4e2..42d70e2 100644 --- a/examples/face_adversarial_attack/example/example_non-target_attack.py +++ b/examples/face_adversarial_attack/example/example_non-target_attack.py @@ -29,7 +29,7 @@ if __name__ == '__main__': targets = AFR.load_data('opencv_photo/target/') - adversarial = AFR.Attack(inputs[0], targets[0]) + adversarial = AFR.FaceAdversarialAttack(inputs[0], targets[0]) attack_method = "non-target attack" adversarial_tensor, mask_tensor = adversarial.train(attack_method)