ARTICLE DETAIL

建站实战干货

来自一线的建站与推广经验沉淀,每一条都经过真实交付验证。

Pytorch 读取t7文件

2026/9/9 15:02:44 拓冰建站 浏览量
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()