pytorch: ground truth similarity matrix

塔_Tass

已于 2024-02-21 18:05:24 修改

阅读量510

点赞数 10

文章标签： pytorch 人工智能 python

于 2024-02-18 20:31:18 首次发布

本文链接：https://blog.csdn.net/weixin_44876302/article/details/136158256

版权

按照真实标签排序pair-wise相似度矩阵的Pytorch代码

本文仅作留档，用于输出可视化

Inputs: Ground-truths $\mathbf{Y}\in\mathbb R^{n\times 1}$ , Similarity matrix $\mathbf{A}\in\mathbb R^{n\times n}$
Outputs: Block diagnal matrix of $\mathbf{A}$ sorted by $\mathbf{Y}$ for visualization.

示例：大规模预训练模型CLIP特征在CIFAR10数据集上的 ground truth similarity matrix
在这里插入图片描述

import torch
import matplotlib.pyplot as plt

Y = labels.argsort()
A = logits.mm(logits.t())
# ground truth similarity matrix
A_sorted = A[Y][:, Y]

# visualize
# A should be normalized to [0,1]
plt.imshow(A_sorted , cmap='Blues', vmin=0, vmax=1)
plt.colorbar()
plt.set_title('Entitle here')
plt.savefig('save_name.png')
plt.close()