使用numpy加载无扩展名文件得到int而非数组,如何正确打开?
解决方法
这个文件不是标准的npy文件,从你读到的torch._utils标识来看,它应该是PyTorch用torch.save()保存的张量文件,直接用numpy加载会解析错误,正确的打开方式如下:
- 使用PyTorch加载,代码示例:
import torch # 加载文件 tensor = torch.load('/opt/data/private/msl/data_thumos/Rgb_TestPJ2_All/video_test_0000006') # 查看张量信息 print(tensor) print(tensor.shape) # 会输出原始形状
补充说明
- 你用
np.load()得到整数的原因:PyTorch保存的文件开头有其特有的序列化魔法数,numpy将其误解析成了整数,和numpy版本无关。 np.fromfile()会忽略文件的格式元信息,所以只能读出一维数组,丢失原始形状,这是正常现象。
内容的提问来源于stack exchange,提问作者muyi miao
相关产品推荐
相关产品推荐

