注意力激活热图的可视化

最新推荐文章于 2024-07-30 13:43:07 发布

代码小白的成长

最新推荐文章于 2024-07-30 13:43:07 发布

阅读量1.5k

点赞数 1

文章标签： python 开发语言

本文链接：https://blog.csdn.net/weixin_43723625/article/details/129425453

版权

本文介绍了如何通过Python将注意力激活热图保存至当前工作目录，并详细展示了运行代码及结果显示的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 图片保存到当前工作目录：

在这里插入图片描述

2. 运行代码

import cv2
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
def visulize_spatial_attention(img_path, attention_mask, ratio=1, cmap="jet"):
    """
    img_path:   image file path to load
    save_path:  image file path to save
    attention_mask: 2-D attention map with np.array type, e.g, (h, w) or (w, h)
    ratio:  scaling factor to scale the output h and w
    cmap:   attention style, default: "jet"
    """
    print