pytorchは.pth形式のファイルをロードします

pytorchが.pthファイルを読み取る方法、およびトーチの.pthファイルを保存用のbinファイルに変換する方法。

import torch
import numpy as np

path_in="in.pth"
path_out="out.pth"

#读取文件
#若安装的pytorch是带Cuda的版本，cuda不能用，则需要指定为cpu运行
data_in = torch.load(path_in, 'cpu')
data_out = torch.load(path_out, 'cpu')
#将torch的数据格式转化为numpy的数据格式
in_data = data_in.numpy().astype(np.uint8)
data_out = data_out.numpy().astype(np.uint8)
#保存文件
in_data.tofile("in.bin")
data_out.tofile("out.bin")