Adds grad-cam support on classifiers

conda/meta.yaml

@@ -27,18 +27,17 @@ requirements:
     - grad-cam {{ grad_cam }}
     - matplotlib {{ matplotlib }}
     - numpy {{ numpy }}
-    - py-opencv {{ py_opencv }}
-    - pandas {{ pandas }}
     - pillow {{ pillow }}
     - psutil {{ psutil }}
     - pytorch {{ pytorch }}
+    - scikit-image {{ scikit_image }}
     - scikit-learn {{ scikit_learn }}
     - scipy {{ scipy }}
     - tabulate {{ tabulate }}
     - torchvision  {{ torchvision }}
     - tqdm {{ tqdm }}
     - tensorboard {{ tensorboard }}
-    - lightning >=2.0.3
+    - lightning {{ lightning }}
     - clapper
   run:
     - python >=3.10
@@ -48,13 +47,14 @@ requirements:
     - {{ pin_compatible('pillow') }}
     - {{ pin_compatible('psutil') }}
     - {{ pin_compatible('pytorch') }}
+    - {{ pin_compatible('scikit-image') }}
     - {{ pin_compatible('scikit-learn') }}
     - {{ pin_compatible('scipy') }}
     - {{ pin_compatible('tabulate') }}
     - {{ pin_compatible('torchvision') }}
     - {{ pin_compatible('tqdm') }}
     - {{ pin_compatible('tensorboard') }}
-    - {{ pin_compatible('lightning') }}
+    - {{ pin_compatible('lightning', max_pin='x.x') }}
     - clapper
 
 test: