pt转换成onnx

要将PyTorch的模型转换为ONNX格式，可以按照以下步骤进行：

1. 安装ONNX和pytorch-to-onnx：

pip install onnx
pip install onnxruntime
pip install git+https://github.com/onnx/pytorch-onnx.git

2. 加载PyTorch模型并将其导出为ONNX格式：

import torch
import torchvision

# 加载PyTorch模型
model = torchvision.models.resnet18()
# 导出为ONNX格式
torch.onnx.export(model,               # PyTorch模型
                  torch.randn(1, 3, 224, 224),  # 输入样本
                  "resnet18.onnx",    # 输出文件名
                  opset_version=11)   # ONNX版本号

3. 可以使用ONNX运行时加载并运行ONNX模型：

import onnxruntime

# 加载ONNX模型
sess = onnxruntime.InferenceSession("resnet18.onnx")

# 准备输入数据
import numpy as np
x = np.random.randn(1, 3, 224, 224).astype(np.float32)

# 运行模型
output = sess.run(None, {"input": x})

# 输出结果
print(output)