Python_pth文件与mat文件的格式转换

最新推荐文章于 2024-03-22 09:53:10 发布

阿妹有点甜

最新推荐文章于 2024-03-22 09:53:10 发布

阅读量616

点赞数 2

分类专栏： Python 文章标签： python 深度学习 pytorch

本文链接：https://blog.csdn.net/m0_59681237/article/details/130370437

版权

Python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

在使用pytorch框架训练神经网络时，网络结构参数会保存为后缀为.pth的文件，我们可以将其提取出来，转换为.mat格式文件。话不多说。直接上代码：

import torch
import scipy.io as sio

pthfile = "pth文件存放路径"

net = torch.load(pthfile,map_location='cpu')

for k in net.keys():
    ValName = k.replace('.','_')
    Value = net[k].cpu().numpy()
    sio.savemat("dataset\{}.mat".format(k),{ValName:Value})