heatmap可视化

最新推荐文章于 2024-04-22 21:48:37 发布

qq_41198463

最新推荐文章于 2024-04-22 21:48:37 发布

阅读量339

点赞数

本文链接：https://blog.csdn.net/qq_41198463/article/details/111372721

版权

本文介绍了如何利用Python的seaborn库创建热力图，详细解释了数据预处理、颜色映射以及如何添加图例和轴标签等步骤，为数据分析中的复杂关系提供直观展示。

摘要由CSDN通过智能技术生成

def restore_rgb_img(batch_img, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]):
    """
    restore the rgb img
    Args:
        batch_img: rgb image with normalization, shape= (batch_size, num_frames, C, H, W)
        mean:      params to restore the rgb image
        std:       params to restore the rgb image

    Returns:

    """
    assert len(batch_img.shape) == 5
    assert batch_img.shape[2] == 3
    C = batch_img.shape[2]
    for channel_index, channle_mean, channel_std in zip(range(C), mean, std):
        batch_img[:, :, [channel_index], :, :] = batch_img[:, :, [channel_index], :, :]*channel_std + channle_mean
    return batch_img


def vis_cos_dist(cos_dist, batch_idx,  width