Update setup.py

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

Author: Beat Buesser

art/evaluations/security_curve/security_curve.py

@@ -23,7 +23,6 @@
 from typing import List, Optional, Tuple, TYPE_CHECKING, Union
 
 import numpy as np
-from matplotlib import pyplot as plt
 
 from art.evaluations.evaluation import Evaluation
 from art.attacks.evasion.projected_gradient_descent.projected_gradient_descent import ProjectedGradientDescent
@@ -162,6 +161,8 @@ def plot(self) -> None:
         Plot the Security Curve of adversarial accuracy as function opf attack budget `eps` together with the accuracy
         on benign samples.
         """
+        from matplotlib import pyplot as plt
+
         plt.plot(self.eps_list, self.accuracy_adv_list, label="adversarial", marker="o")
         plt.plot([self.eps_list[0], self.eps_list[-1]], [self.accuracy, self.accuracy], linestyle="--", label="benign")
         plt.legend()

setup.py

@@ -22,7 +22,7 @@
     "sphinx-autodoc-annotation",
     "sphinx-autodoc-typehints",
     "matplotlib",
-    "numpy",
+    "numpy>=1.18.0",
     "scipy>=1.4.1",
     "six>=1.13.0",
     "scikit-learn>=0.22.2,<0.24.3",
@@ -65,13 +65,13 @@ def get_version(rel_path):
         "lightgbm": ["lightgbm"],
         "mxnet": ["mxnet"],
         "tensorflow": ["tensorflow", "tensorflow_addons", "h5py"],
-        "tensorflow_image": ["tensorflow", "tensorflow_addons", "h5py", "Pillow", "ffmpeg-python"],
-        "tensorflow_audio": ["tensorflow", "tensorflow_addons", "h5py", "pydub", "resampy"],
+        "tensorflow_image": ["tensorflow", "tensorflow_addons", "h5py", "Pillow", "ffmpeg-python", "opencv-python"],
+        "tensorflow_audio": ["tensorflow", "tensorflow_addons", "h5py", "pydub", "resampy", "librosa"],
         "pytorch": ["torch", "torchvision"],
-        "pytorch_image": ["torch", "torchvision", "kornia", "Pillow", "ffmpeg-python"],
-        "pytorch_audio": ["torch", "torchvision", "torchaudio", "pydub", "resampy"],
+        "pytorch_image": ["torch", "torchvision", "kornia", "Pillow", "ffmpeg-python", "opencv-python"],
+        "pytorch_audio": ["torch", "torchvision", "torchaudio", "pydub", "resampy", "librosa"],
         "xgboost": ["xgboost"],
-        "lingvo_asr": ["tensorflow-gpu==2.1.0", "lingvo==0.6.4"],
+        "lingvo_asr": ["tensorflow-gpu==2.1.0", "lingvo==0.6.4", "pydub", "resampy", "librosa"],
         "all": [
             "mxnet",
             "catboost",
@@ -91,6 +91,8 @@ def get_version(rel_path):
             "resampy",
             "ffmpeg-python",
             "cma",
+            "librosa",
+            "opencv-python",
         ],
     },
     classifiers=[