Trusted-AI / adversarial-robustness-toolbox

Adversarial Robustness Toolbox (ART) - Python Library for Machine Learning Security - Evasion, Poisoning, Extraction, Inference - Red and Blue Teams

Home Page:https://adversarial-robustness-toolbox.readthedocs.io/en/latest/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Implement `get_activations()` for PyTorch Object Detectors

f4str opened this issue · comments

Is your feature request related to a problem? Please describe.
The current PyTorch object detectors (PyTorchObjectDetector, PyTorchFasterRCNN, PyTorchYolo, PyTorchDetectionTransformer) do not implement the get_activations() method and therefore cannot extract activations.

Describe the solution you'd like
Using the model wrapper method used by HuggingFaceClassifierPyTorch, support can be added for activations for the PyTorch object detectors. This will allow usage of activation-based defenses such as Activation Clustering and Spectral Signatures.

Describe alternatives you've considered
N/A

Additional context
This will be worked on after #2267 is completed to avoid redundant code.

Hi @f4str Sounds good!