python .npy file的读写等

1.读写.npy文件

• .npy文件是numpy专用的二进制文件

import numpy as np
arr=np.array([[1,2,3],[4,5,6]])
np.save("weight.npy",arr)  #将数据arr保存到weight.npy文件中
loadData=np.load('weight.npy')  #加载npy文件
#如果加载一下别的.npy文件出现如下错误
#   raise ValueError("Object arrays cannot be loaded when "
#ValueError: Object arrays cannot be loaded when allow_pickle=False
#需要加入参数
#loadData=np.load('weight.npy',allow_pickle=True)
print(type(loadData))
print(loadData.shape)
print(loadData)

• 实战加载自己的.npy文件

import numpy as np
pre_train=np.load("vgg16_no_fc.npy",allow_pickle=True,encoding="latin1") #加载预先训练的权重,可以是.npy文件
data_dic=pre_train.item() #item()方法把字典中每对key和value组成一个元组，并把这些元组放到列表中返回
print(type(data_dic))
#print(data_dic)  查看字典的键和值
print(data_dic['conv5_1']['kernel'])  #通过键索引取值
print(type(data_dic['conv5_1']['kernel']))
print(data_dic['conv5_1']['kernel'].shape) #查看形状

参考