Python读取npy文件怎样实现,代码是什么
Admin 2022-11-03 群英技术资讯 625 次浏览
注:.npy文件是numpy专用的二进制文件。
import numpy as np arr = np.array([[1, 2, 3], [4, 5, 6]]) np.save('weight.npy', arr) loadData = np.load('weight.npy') print("----type----") print(type(loadData)) print("----shape----") print(loadData.shape) print("----data----") print(loadData)
至于具体API参见:https://docs.scipy.org/doc/numpy/index.html
在深度神经网络训练过程中通常需要读取预训练权重,预训练权重通常是 .npy文件,比如vgg16.npy(https://pan.baidu.com/s/1Ru5FJVSPjYTHZwlmzRwRvQ 提取码:ygxw)。本次就以分析vgg16.npy为例进行说明。
import numpy as np # 注意编码方式 pre_train = np.load("vgg16.npy", allow_pickle=True, encoding="latin1") print("------type-------") print(type(pre_train)) print("------shape-------") print(pre_train.shape) print("------data-------") print(pre_train)
这是个啥?为啥shape没有? 但是可以看出来 pre_train 里元素应该是一个字典,我们尝试取出来。
注:ndarray.item()是复制数组中的一个元素,并将其返回。具体语法参见:https://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.item.html?highlight=item#numpy.ndarray.item
import numpy as np pre_train = np.load("vgg16.npy", allow_pickle=True, encoding="latin1") data_dic = pre_train.item() print("------type-------") print(type(data_dic)) print("------conv1_1 data-------") print(data_dic['conv1_1']) # 返回一个列表,该列表有两个array,表示conv1_1的权重w与偏置b print("------conv1_1 shape-------") print((data_dic['conv1_1'][0]).shape)
看看结果:
可以发现,这是第一个卷积层的权重参数,输入channel是3,输出channel是64。
numpy中的二进制文件的读写:
save
np.save ("./文件名", 数组名):以二进制的格式保存数据
load
np.load("./文件名.npy"): 函数是从二进制的文件中读取数据
savez
np.savez(’./文件名’,数组名1,数组名2,…):savez 函数可以将多个数组保存到一个文件中
(1)save操作
import numpy as np a=np.arange(5) np.save('get.npy',a)
(2)load操作
import numpy as np a=np.load('load.npy') print(a)
(3)savez操作
import numpy as np a=np.arange(3) b=np.arange(4) c=np.arange(5) np.savez('array_save.npz',a,b,c) 多个ndarray类型的数组
到此,关于“Python读取npy文件怎样实现,代码是什么”的学习就结束了,希望能够解决大家的疑惑,另外大家动手实践也很重要,对大家加深理解和学习很有帮助。如果想要学习更多的相关知识,欢迎关注群英网络,小编每天都会给大家分享实用的文章!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。
猜你喜欢
Python 中 global 关键字可以定义一个变量为全局变量,但是这个仅限于在一个模块(py文件)中调用全局变量,在另外一个py文件 再次使用 global x 也是无法访问到的,这篇文章主要介绍了Python跨文件全局变量的使用,需要的朋友可以参考下
这篇文章主要为大家介绍了Python进行密码学反向密码的教程详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
本文主要介绍了opencv+tesseract实现验证码识别的示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
进程,一个新鲜的字眼,可能有些人并不了解,它是系统某个运行程序的载体,这个程序可以有单个或者多个进程,一般来说,进程是通过系统CPU 内核数来分配并设置的,我们可以来看下系统中的进程
这篇文章主要介绍了Python项目的服务器部署,帮助大家更好的理解和学习使用python,感兴趣的朋友可以了解下
成为群英会员,开启智能安全云计算之旅
立即注册Copyright © QY Network Company Ltd. All Rights Reserved. 2003-2020 群英 版权所有
增值电信经营许可证 : B1.B2-20140078 粤ICP备09006778号 域名注册商资质 粤 D3.1-20240008