Anatomy-Aware-Classification-axSpA

Running on Zero

FJDorfner commited on May 19

Commit

0695ece

•

1 Parent(s): fbff24b

Upload Model_Class.py

Files changed (1) hide show

Model_Class.py CHANGED Viewed

@@ -94,7 +94,7 @@ def make_GradCAM(image, device):
         aug_smooth=False,
         eigen_smooth=True,
     )
-    grayscale_cam = grayscale_cam.to('cpu').squeeze()
     jet = plt.colormaps.get_cmap("inferno")
     newcolors = jet(np.linspace(0, 1, 256))

         aug_smooth=False,
         eigen_smooth=True,
     )
+    grayscale_cam = grayscale_cam.squeeze()
     jet = plt.colormaps.get_cmap("inferno")
     newcolors = jet(np.linspace(0, 1, 256))