从Pytorch模型pth文件中读取参数成numpy矩阵的操作

更新时间：2021年03月04日 14:46:27 作者：木盏

这篇文章主要介绍了从Pytorch模型pth文件中读取参数成numpy矩阵的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

目的：

把训练好的pth模型参数提取出来，然后用其他方式部署到边缘设备。

Pytorch给了很方便的读取参数接口：

nn.Module.parameters()

直接看demo：

from torchvision.models.alexnet import alexnet 
model = alexnet(pretrained=True).eval().cuda()
parameters = model.parameters()
for p in parameters:
  numpy_para = p.detach().cpu().numpy()
  print(type(numpy_para))
  print(numpy_para.shape)

上面得到的numpy_para就是numpy参数了~

Note：

model.parameters()是以一个生成器的形式迭代返回每一层的参数。所以用for循环读取到各层的参数，循环次数就表示层数。

而每一层的参数都是torch.nn.parameter.Parameter类型，是Tensor的子类，所以直接用tensor转numpy（即p.detach().cpu().numpy()）的方法就可以直接转成numpy矩阵。

方便又好用，爆赞~

补充：pytorch训练好的.pth模型转换为.pt

将python训练好的.pth文件转为.pt

import torch
import torchvision
from unet import UNet
model = UNet(3, 2)#自己定义的网络模型
model.load_state_dict(torch.load("best_weights.pth"))#保存的训练模型
model.eval()#切换到eval（）
example = torch.rand(1, 3, 320, 480)#生成一个随机输入维度的输入
traced_script_module = torch.jit.trace(model, example)
traced_script_module.save("model.pt")

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。

您可能感兴趣的文章:

在python3中pyqt5和mayavi不兼容问题的解决方法
今天小编就为大家分享一篇在python3中pyqt5和mayavi不兼容问题的解决方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-01-01
Python机器学习NLP自然语言处理基本操作关键词
本文是Python机器学习NLP自然语言处理系列文章，带大家开启一段学习自然语言处理 (NLP) 的旅程. 本文主要学习NLP自然语言处理关键词的操作
2021-09-09
python3基于TCP实现CS架构文件传输
这篇文章主要为大家详细介绍了python3基于TCP实现CS架构文件传输，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-07-07
如何查看Mac本机的Python3安装路径
这篇文章主要介绍了如何查看Mac本机的Python3安装路径问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-03-03
Jupyter Notebook远程登录及密码设置操作
今天小编就为大家分享一篇Jupyter Notebook远程登录及密码设置操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-04-04
Python 爬虫爬取指定博客的所有文章
因为Google App Engine 被墙，我无法继续完善我的Moven project 还有20＋天才回去，怕到时候会忘记project的进度和细节就趁着个冷的什么都不想干的时候，大概的总结一下
2016-02-02
Python实现单例模式的四种方式详解
单例模式可以保证一个类仅有一个实例，并提供一个访问它的全局访问点。本文为大家介绍了Python实现单例模式的四种方式，需要的可以参考一下
2022-05-05
Python的Flask框架中@app.route的用法教程
这篇文章主要介绍了Python的Flask框架中@app.route的用法教程,包括相关的正则表达式讲解，是Flask学习过程当中的基础知识，需要的朋友可以参考下
2015-03-03
django美化后台django-suit的安装配置操作
这篇文章主要介绍了django美化后台django-suit的安装配置操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-07-07
Python更改pip镜像源的方法示例
这篇文章主要介绍了Python更改pip镜像源的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-12-12