Browse Source

update examples/community/face_adversarial_attack/example_target_attack.py.

Signed-off-by: 君君臣臣君 <mingjun@isrc.iscas.ac.cn>
pull/418/head
君君臣臣君 Gitee 2 years ago
parent
commit
61fc36d3ad
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 matplotlib.image as mp
from mindspore import context
import AFR
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 = "target_attack"



Loading…
Cancel
Save