fix merge, lower gif size

jacobgil · May 28, 2024 · b9f477e · b9f477e
1 parent 3f6b14d
commit b9f477e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/examples/multiorgan_segmentation.gif b/examples/multiorgan_segmentation.gif
diff --git a/pytorch_grad_cam/utils/image.py b/pytorch_grad_cam/utils/image.py
@@ -165,10 +165,11 @@ def scale_cam_image(cam, target_size=None):
         img = img - np.min(img)
         img = img / (1e-7 + np.max(img))
         if target_size is not None:
-        if len(img.shape) > 3:
-            img = zoom(np.float32(img), [(t_s/i_s) for i_s, t_s in zip(img.shape, target_size[::-1])])
-        else:
-            img = cv2.resize(np.float32(img), target_size)
+            if len(img.shape) > 3:
+                img = zoom(np.float32(img), [
+                           (t_s / i_s) for i_s, t_s in zip(img.shape, target_size[::-1])])
+            else:
+                img = cv2.resize(np.float32(img), target_size)
 
         result.append(img)
     result = np.float32(result)

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name='grad-cam',
-    version='1.5.0',
+    version='1.5.2',
     author='Jacob Gildenblat',
     author_email='jacob.gildenblat@gmail.com',
     description='Many Class Activation Map methods implemented in Pytorch for classification, segmentation, object detection and more',