npy文件的打开，读取

在复现Sketchmate的时候，碰到了一个叫npy文件的东西：

alpha = 0.001
cf_class = torch.from_numpy(np.load("center_feature_ssn.npy"))
cf_class = cf_class.to(device=device, dtype=torch.float32)

torch.from_numpy()方法很好理解，就是把数组转换成张量。本来想看下这个文件的内容，但是怎么都没有打开这个.npy文件的软件。后来得知可以通过简单的load和print打印输出就可以进行读取和查看npy文件内容，现在记录一下：

# 导入所需的包
import torch
import numpy as np

# 导入npy文件路径位置
cf_class = torch.from_numpy(np.load("center_feature_ssn.npy"))

print(cf_class) # 输出文件
print(cf_class.shape) # 输出张量形状
print(type(cf_class)) # 输出格式

输出：

tensor([[ 0.0000e+00,  8.4585e-03,  0.0000e+00,  ...,  9.3659e+01,
         -4.3256e+01, -2.0462e+01],
        [ 0.0000e+00,  1.0637e-02,  0.0000e+00,  ...,  8.8363e+01,
         -4.4654e+01, -1.8992e+01],
        [ 0.0000e+00,  2.2176e-02,  0.0000e+00,  ...,  9.5005e+01,
         -4.1959e+01, -2.2259e+01],
        ...,
        [ 0.0000e+00,  1.0582e-02,  0.0000e+00,  ...,  9.1089e+01,
         -4.5272e+01, -1.7538e+01],
        [ 0.0000e+00,  3.6281e-02,  0.0000e+00,  ...,  9.4309e+01,
         -4.2236e+01, -2.1581e+01],
        [ 0.0000e+00,  1.0220e-01,  0.0000e+00,  ...,  7.3616e+01,
         -5.1074e+01, -1.1483e+01]])
torch.Size([40, 2088])
<class 'torch.Tensor'>

表明里面存储了一个（40，2088）形状的张量。

.npy文件是numpy专用的二进制文件。在使用训练好的模型时，npy文件可以保存模型。npy文件也可以读取预训练权重。

附：Sketchmate复现源码参考：

tosmaster/imagevision (github.com)https://github.com/tosmaster/imagevision#readme

npy文件的打开，读取

猜你喜欢