Fix cv2.resize issue (#494)

jacobgil · May 28, 2024 · f0371ab · f0371ab
1 parent 51ae192
commit f0371ab
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
@@ -163,7 +163,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:
-            img = cv2.resize(img, target_size)
+            img = cv2.resize(np.float32(img), target_size)
         result.append(img)
     result = np.float32(result)