Fix 3D

jacobgil · Aug 18, 2024 · 7ff4d2e · 7ff4d2e
1 parent 1610034
commit 7ff4d2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_grad_cam/utils/image.py b/pytorch_grad_cam/utils/image.py
@@ -165,7 +165,7 @@ 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:
+            if len(img.shape) > 2:
                 img = zoom(np.float32(img), [
                            (t_s / i_s) for i_s, t_s in zip(img.shape, target_size[::-1])])
             else: