Skip to content
Snippets Groups Projects
Commit e7169bdb authored by ogueler@idiap.ch's avatar ogueler@idiap.ch Committed by Daniel CARRON
Browse files

added dependancies

parent 174f2ccf
No related branches found
No related tags found
1 merge request!12Adds grad-cam support on classifiers
...@@ -24,8 +24,11 @@ requirements: ...@@ -24,8 +24,11 @@ requirements:
- python >=3.10 - python >=3.10
- pip - pip
- click {{ click }} - click {{ click }}
- grad-cam >=1.4.8
- matplotlib {{ matplotlib }} - matplotlib {{ matplotlib }}
- numpy {{ numpy }} - numpy {{ numpy }}
- opencv {{ opencv }}
- pandas {{ pandas }}
- pillow {{ pillow }} - pillow {{ pillow }}
- psutil {{ psutil }} - psutil {{ psutil }}
- pytorch {{ pytorch }} - pytorch {{ pytorch }}
......
...@@ -30,6 +30,7 @@ dependencies = [ ...@@ -30,6 +30,7 @@ dependencies = [
"clapper", "clapper",
"click", "click",
"numpy", "numpy",
"pandas",
"scipy", "scipy",
"scikit-learn", "scikit-learn",
"tqdm", "tqdm",
...@@ -37,11 +38,13 @@ dependencies = [ ...@@ -37,11 +38,13 @@ dependencies = [
"tabulate", "tabulate",
"matplotlib", "matplotlib",
"pillow", "pillow",
"opencv",
"torch>=1.8", "torch>=1.8",
"torchvision>=0.10", "torchvision>=0.10",
"lightning>=2.0.3", "lightning>=2.0.3",
"pydantic <2.0,>=1.7.4", # temporary, until issue #31 is fixed "pydantic <2.0,>=1.7.4", # temporary, until issue #31 is fixed
"tensorboard", "tensorboard",
"grad-cam>=1.4.8",
] ]
[project.urls] [project.urls]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment