.npy文件的打开方式
程序员文章站
2022-06-04 08:10:48
...
1. 在深度学习领域,会训练出一个模型,在使用训练好的模型时,其中有一种保存的模型文件格式叫.npy。如何打开呢?
2. 实现代码即可打开:
import numpy as np
test=np.load('./bvlc_alexnet.npy',encoding = "latin1") #加载文件
doc = open('1.txt', 'a') #打开一个存储文件,并依次写入
print(test, file=doc) #将打印内容写入文件中
3. 模型文件(.npy)解析:
import numpy as np
from numpy import * #使用numpy的属性且不需要在前面加上numpy
import tensorflow as tf
#模型文件(.npy)部分内容如下:由一个字典组成,字典中的每一个键对应一层网络模型参数。(包括权重w和偏置b)
a = {'conv1':[array([[1,2],[3,4]],dtype=float32),array([5,6],dtype=float32)],'conv2':[array([[1,2],[3,4]],dtype=float32),array([5,6],dtype=float32)]}
conv1_w = a['conv1'][0]
conv1_b = a['conv1'][1]
conv2_w = a['conv2'][0]
conv2_b = a['conv2'][1]
print(conv1_w)
print(tf.Variable(conv1_w))
print(conv1_b)
print(tf.Variable(conv1_b))
4. 结果:
[[ 1. 2.]
[ 3. 4.]]
<tf.Variable 'Variable:0' shape=(2, 2) dtype=float32_ref>
[ 5. 6.]
<tf.Variable 'Variable_1:0' shape=(2,) dtype=float32_ref>
原文链接:https://blog.csdn.net/william_hehe/article/details/80412227
上一篇: Docker基础教程