set rbg order in overlaying images

reginabarzilaygroup · Feb 7, 2024 · 9d15789 · 9d15789
1 parent 0477675
commit 9d15789
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sybil/utils/visualization.py b/sybil/utils/visualization.py
@@ -51,9 +51,9 @@ def visualize_attentions(
         overlayed_images = []
         for i in range(N):
             overlayed = np.zeros((512, 512, 3))
-            overlayed[..., 0] = images[i]
+            overlayed[..., 2] = images[i]
             overlayed[..., 1] = images[i]
-            overlayed[..., 2] = np.clip(
+            overlayed[..., 0] = np.clip(
                 (attention_up[0, 0, i] * gain * 256) + images[i],
                 a_min=0,
                 a_max=256,