refactor: misc

Author: Lodour

requirements.txt

@@ -15,6 +15,7 @@ loguru
 IPython
 pytest
 prettytable
+imagesize
 
 # api
 tencentcloud-sdk-python

scripts/attack_blackbox.py

@@ -143,7 +143,7 @@ def parse_args():
     _('--no-smart-median', action='store_true', help='disable gradient-efficient median approximation')
     _('--precise-noise', action='store_true', help='use the straightforward scaling-aware noise sampling')
     # Misc
-    _('-o', '--output', type=Path, default='static/blackbox-results', help='path to output directory')
+    _('-o', '--output', type=Path, default='static/logs', help='path to output directory')
     _('-t', '--tag', default='', type=str, help='name for this experiment')
     _('-g', '--gpu', default=0, type=int, help='GPU id')
     args = parser.parse_args()

src/attacks/sign_opt.py

@@ -151,8 +151,8 @@ def generate(self, x0: np.ndarray, y0: int, tag=None):
 
                 # logging
                 nb_query += grad_queries + ls_count
-                pbar.set_postfix({'query': nb_query, 'l2': f'{gg:.3f}'})
-                self.log.append((i, nb_query, gg))
+                pbar.set_postfix({'query': nb_query, 'l2': f'{gg:.5f}'})
+                self.log.append((nb_query, gg))
 
                 # stop if we reach the query limit
                 if nb_query > self.max_query:

src/defenses/prevention.py

@@ -134,9 +134,9 @@ def __init__(self, *args, **kwargs):
     def pooling(self, x: torch.Tensor) -> torch.Tensor:
         x = self.unfold(x)
 
-        low = x.quantile(0.3, dim=-1)[..., None]
+        low = x.quantile(0.2, dim=-1)[..., None]
         med = x.median(dim=-1).values[..., None]
-        high = x.quantile(0.7, dim=-1)[..., None]
+        high = x.quantile(0.8, dim=-1)[..., None]
         ind = (low <= x) * (x <= high) * (1 - (x - med).abs())
 
         out = (x * ind).sum(dim=-1) / ind.sum(dim=-1)