欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

.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