tensor类型，单通道feature画注意力图

本文主要是介绍tensor类型，单通道feature画注意力图，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

import numpy as np
import matplotlib.pyplot as plt

def normalization(data):
    rang=np.max(data)-np.min(data)
    return (data-np.min(data))/rang


def save_1_feature_map(feature_map):
    # [1, H, W] -> [H, W]
    im = np.squeeze(feature_map.detach().cpu().numpy())

    # normalization
    im=normalization(-im)

    # plt.figure()
    plt.axis('off')
    plt.imshow(im, cmap='jet')
    plt.savefig('feature_map{}.png'.format(1))

这篇关于tensor类型，单通道feature画注意力图的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！