python中h5文件和pt文件

最新推荐文章于 2024-08-19 08:23:16 发布

负能量小公举，

最新推荐文章于 2024-08-19 08:23:16 发布

阅读量2.4w

点赞数 26

文章标签： pytorch

本文链接：https://blog.csdn.net/weixin_43526420/article/details/110321466

版权

本文介绍了h5文件的结构，其中包含组“group”和数据集“dataset”，类似于Linux文件系统。pt文件是PyTorch模型的保存格式，.pt, .pth, .pkl的区别仅在于个人喜好，实际内容无异。文中展示了两种模型的保存和调用方法：一种保存权重参数，另一种保存整个模型状态。" 130374956,13222000,Python OpenCV 实现巴特沃斯低通滤波图像处理,"['计算机视觉', '图像处理', 'Python编程', 'OpenCV库', '滤波技术']

摘要由CSDN通过智能技术生成

python中h5文件和pt文件

- h5文件
- pt文件

h5文件

h5文件中有两个核心的概念：组“group”和数据集“dataset”。一个h5文件就是 “dataset” 和 “group” 二合一的容器。
dataset ：简单来讲类似数组组织形式的数据集合，像 numpy 数组一样工作，一个dataset即一个numpy.ndarray。具体的dataset可以是图像、表格，甚至是pdf文件和excel。
group：包含了其它 dataset(数组) 和其它 group ，像字典一样工作。
一个h5文件被像linux文件系统一样被组织起来：dataset是文件，group是文件夹，它下面可以包含多个文件夹(group)和多个文件(dataset)。
形象来看h5数据组织方式大概像酱婶儿的，诺！跟文件系统一样，大概知道它为啥叫层次数据格式了吧！
h5文件root qroup

pt文件

我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件，这几种模型文件在格式上有什么区别吗？其实它们并不是在格式上有区别，只是后缀不同而已（仅此而已），在用torch.save()函数保存模型文件时，各人有不同的喜好，有些人喜欢用.pt后缀，有些人喜欢用.pth或.pkl.用相同的torch.save（）语句保存出来的模型文件没有什么不同。
在pytorch官方的文档/代码里，有用.pt的