Browse Source

update examples/community/face_adversarial_attack/example_non-target_attack.py.

Signed-off-by: 君君臣臣君 <mingjun@isrc.iscas.ac.cn>
pull/425/head
君君臣臣君 Gitee 2 years ago
parent
commit
aeb56eef56
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      examples/community/face_adversarial_attack/example_non-target_attack.py

+ 4
- 4
examples/community/face_adversarial_attack/example_non-target_attack.py View File

@@ -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)


Loading…
Cancel
Save