mask array flatten

fluid/adversarial/advbox/attacks/saliency.py

@@ -63,7 +63,7 @@ class SaliencyMapAttack(Attack):
                     return adversary
 
                 # stop if mask is all zero
-                if not any(mask):
+                if not any(mask.flatten()):
                     return adversary
 
                 logging.info('step = {}, original_label = {}, adv_label={}'.
@@ -105,9 +105,6 @@ class SaliencyMapAttack(Attack):
         target_labels = [t for t in target_labels if t != original_label]
         target_labels = target_labels[:num_random_target]
 
-        # str_target_labels = [str(t) for t in target_labels]
-        # logging.info('Random target labels: {}'.format(', '.join(str_target_labels)))
-
         return target_labels
 
     def _saliency_map(self, image, target, labels, mask, fast=False):