Skip to content
Snippets Groups Projects

Adds grad-cam support on classifiers

Merged André Anjos requested to merge add-datamodule-gradcam into main
Compare and Show latest version
18 files
+ 174
745
Compare changes
  • Side-by-side
  • Inline
Files
18
+ 4
0
@@ -30,6 +30,7 @@ requirements:
- pillow {{ pillow }}
- psutil {{ psutil }}
- pytorch {{ pytorch }}
- scikit-image {{ scikit_image }}
- scikit-learn {{ scikit_learn }}
- scipy {{ scipy }}
- tabulate {{ tabulate }}
@@ -37,6 +38,7 @@ requirements:
- tqdm {{ tqdm }}
- tensorboard {{ tensorboard }}
- lightning {{ lightning }}
- lightning >=2.1.0,!=2.1.3
- clapper
run:
- python >=3.10
@@ -46,6 +48,7 @@ requirements:
- {{ pin_compatible('pillow') }}
- {{ pin_compatible('psutil') }}
- {{ pin_compatible('pytorch') }}
- {{ pin_compatible('scikit-image') }}
- {{ pin_compatible('scikit-learn') }}
- {{ pin_compatible('scipy') }}
- {{ pin_compatible('tabulate') }}
@@ -53,6 +56,7 @@ requirements:
- {{ pin_compatible('tqdm') }}
- {{ pin_compatible('tensorboard') }}
- {{ pin_compatible('lightning', max_pin='x.x') }}
- lightning >=2.1.0,!=2.1.3
- clapper
test:
Loading