From e21634aee27f336df472762249a8ea41152b5790 Mon Sep 17 00:00:00 2001
From: ljhOfGithub <42089575+ljhOfGithub@users.noreply.github.com>
Date: Fri, 14 Jul 2023 16:53:27 +0800
Subject: [PATCH] fix:score2percentile()

---
 vis_utils/heatmap_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vis_utils/heatmap_utils.py b/vis_utils/heatmap_utils.py
index 1dc06d66..c8fc0e8b 100644
--- a/vis_utils/heatmap_utils.py
+++ b/vis_utils/heatmap_utils.py
@@ -20,7 +20,7 @@
 device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def score2percentile(score, ref):
-    percentile = percentileofscore(ref, score)
+    percentile = percentileofscore(ref.squeeze(), score.squeeze())
     return percentile
 
 def drawHeatmap(scores, coords, slide_path=None, wsi_object=None, vis_level = -1, **kwargs):