提交 2bf99129 编写于 作者: wgzqz's avatar wgzqz

No need to reshape the original shape.

上级 6c39bce4
...@@ -45,7 +45,7 @@ class DeepFoolAttack(Attack): ...@@ -45,7 +45,7 @@ class DeepFoolAttack(Attack):
labels = np.arange(class_count) labels = np.arange(class_count)
gradient = self.model.gradient([(adversary.original, pre_label)]) gradient = self.model.gradient([(adversary.original, pre_label)])
x = adversary.original.reshape(gradient.shape) x = adversary.original
for iteration in xrange(iterations): for iteration in xrange(iterations):
w = np.inf w = np.inf
w_norm = np.inf w_norm = np.inf
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册