pth模型转换为onnx

将PyTorch模型转换为ONNX格式的代码如下：

import torch
import torch.onnx as onnx
import torchvision.models as models

# 定义输入数据
example_input = torch.randn(1, 3, 224, 224)

# 加载PyTorch模型
model = models.resnet18(pretrained=True)

# 设置模型为eval模式
model.eval()

# 导出ONNX模型
onnx.export(model, example_input, "resnet18.onnx", verbose=True)

在这个例子中，我们首先定义了一个example_input作为输入数据，然后加载了一个预训练的ResNet-18模型。在导出ONNX模型之前，我们需要将模型设置为eval模式，以确保模型的输出是确定的。最后，我们使用torch.onnx.export函数将模型导出为ONNX格式，并指定导出的文件名为"resnet18.onnx"。