File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
art/attacks/evasion/adversarial_patch Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 41
41
42
42
import torch
43
43
44
- from art .utils import CLASSIFIER_NEURALNETWORK_TYPE
44
+ from art .utils import CLASSIFIER_NEURALNETWORK_TYPE , PYTORCH_OBJECT_DETECTOR_TYPE
45
45
46
46
logger = logging .getLogger (__name__ )
47
47
@@ -72,7 +72,7 @@ class AdversarialPatchPyTorch(EvasionAttack):
72
72
73
73
def __init__ (
74
74
self ,
75
- estimator : "CLASSIFIER_NEURALNETWORK_TYPE" ,
75
+ estimator : "CLASSIFIER_NEURALNETWORK_TYPE | PYTORCH_OBJECT_DETECTOR_TYPE " ,
76
76
rotation_max : float = 22.5 ,
77
77
scale_min : float = 0.1 ,
78
78
scale_max : float = 1.0 ,
@@ -91,7 +91,7 @@ def __init__(
91
91
"""
92
92
Create an instance of the :class:`.AdversarialPatchPyTorch`.
93
93
94
- :param estimator: A trained estimator.
94
+ :param estimator: A trained PyTorch estimator for classification or object detection .
95
95
:param rotation_max: The maximum rotation applied to random patches. The value is expected to be in the
96
96
range `[0, 180]`.
97
97
:param scale_min: The minimum scaling applied to random patches. The value should be in the range `[0, 1]`,
You can’t perform that action at this time.
0 commit comments