fixed attention map inversion

Ankit Gupta · Ankit Gupta · commit f5fe9f92da66 · 2025-03-13T14:49:53.000+01:00
diff --git a/inference.py b/inference.py
@@ -91,7 +91,6 @@ def save_attention_map(attn, input_shape, output_path):
         scale_each=True,
     )
     attn = (attn.permute(1, 2, 0).cpu().numpy() * 255).astype(np.uint8)
-    attn = 255 - attn
     imsave(output_path + "_attention_map.png", attn)
 
 

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,6 @@ def save_attention_map(attn, input_shape, output_path):`
`91`	`91`	`scale_each=True,`
`92`	`92`	`)`
`93`	`93`	`attn = (attn.permute(1, 2, 0).cpu().numpy() * 255).astype(np.uint8)`
`94`		`- attn = 255 - attn`
`95`	`94`	`imsave(output_path + "_attention_map.png", attn)`
`96`	`95`
`97`	`96`