本文主要是介绍Pytorch 读取t7文件,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
Pytorch 1.0以上可以使用:
import torchfileth_path = r"./path/xx.t7"
data = torchfile.load(th_path)print(data.shape)
若data的尺寸为0,则将torch版本降为0.4.1,并使用以下函数:
from torch.utils.serialization import load_luath_path = r"./path/xx.t7"
data = load_lua(th_path).numpy()print(data.shape)
注意:
若是在Windows的系统中读取t7文件,一定要记得要用long_size=8
:
data = torchfile.load(th_path,long_size=8)
或
data = load_lua(th_path,long_size=8).numpy()
这篇关于Pytorch 读取t7文件的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!