Browse Source

update examples/community/face_adversarial_attack/example_target_attack.py.

Signed-off-by: 君君臣臣君 <mingjun@isrc.iscas.ac.cn>
pull/425/head
君君臣臣君 Gitee 2 years ago
parent
commit
00992f07ec
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_target_attack.py

+ 4
- 4
examples/community/face_adversarial_attack/example_target_attack.py View File

@@ -16,7 +16,7 @@
import numpy as np import numpy as np
import matplotlib.image as mp import matplotlib.image as mp
from mindspore import context from mindspore import context
import AFR
import adversarial_attack


context.set_context(mode=context.GRAPH_MODE, device_target="GPU") 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__': 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 = "target_attack" attack_method = "target_attack"




Loading…
Cancel
Save