Fix dependencies

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

tests/attacks/evasion/test_overload_attack.py

@@ -42,11 +42,11 @@ def test_generate(art_warning):
         # torch.serialization.add_safe_globals([torch.nn.modules.activation.SiLU])
         # torch.serialization.add_safe_globals([torch.nn.modules.conv.Conv2d])
         # torch.serialization.add_safe_globals([torch.nn.modules.upsampling.Upsample])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.DetectionModel])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.SPPF])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.DetectionModel])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.SPPF])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
         # torch.serialization.add_safe_globals([Conv])
         # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
         # torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
@@ -103,11 +103,11 @@ def test_check_params(art_warning):
         # torch.serialization.add_safe_globals([torch.nn.modules.activation.SiLU])
         # torch.serialization.add_safe_globals([torch.nn.modules.conv.Conv2d])
         # torch.serialization.add_safe_globals([torch.nn.modules.upsampling.Upsample])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.DetectionModel])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.SPPF])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
-        # torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.DetectionModel])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.SPPF])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
+        torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
         # torch.serialization.add_safe_globals([Conv])
         # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
         # torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])