pytorch将.pth文件里的参数保存为.mat

import torch
import scipy.io as sio



pthfile = r'D:\LY\DLmodify\3Deblur_cnn\Deblur_cvpr\document\3test\modelDict\test_pycharm.pth'  # .pth文件地址
net = torch.load(pthfile) 

#print(type(net))  # 类型是 dict
print(len(net))   # 长度为 8，即存在8个 key-value 键值对


for k in net.keys():
	s = k.split('.')
	#print(s[0])
	s = '{}_{}'.format(s[0],s[1][0])
	s1 = net[k].cpu().numpy()
	#print(s)
	#print(s1)
	sio.savemat("model\{}.mat".format(s), {"array": s1})