diff --git a/conda/meta.yaml b/conda/meta.yaml index fd2999083224f0e51939c677240c9ac626614b9f..887572cc5684a23fb627dc013e757cf3c89ad32b 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -24,8 +24,11 @@ requirements: - python >=3.10 - pip - click {{ click }} + - grad-cam >=1.4.8 - matplotlib {{ matplotlib }} - numpy {{ numpy }} + - opencv {{ opencv }} + - pandas {{ pandas }} - pillow {{ pillow }} - psutil {{ psutil }} - pytorch {{ pytorch }} diff --git a/pyproject.toml b/pyproject.toml index 16082e2d0c8a57b44d3d90a7aa417be960b78227..106c786f4f6736b1ae628061d896f10a6b0906bb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,6 +30,7 @@ dependencies = [ "clapper", "click", "numpy", + "pandas", "scipy", "scikit-learn", "tqdm", @@ -37,11 +38,13 @@ dependencies = [ "tabulate", "matplotlib", "pillow", + "opencv", "torch>=1.8", "torchvision>=0.10", "lightning>=2.0.3", "pydantic <2.0,>=1.7.4", # temporary, until issue #31 is fixed "tensorboard", + "grad-cam>=1.4.8", ] [project.urls]