将pytorch模型转换为onnx

Question

我正在尝试将 pytorch 模型转换为 onnx 模型。我有 pth 模型，我加载模型并将其导出到 onnx。输入是具有 3 个通道的 512x 512 图像。该模型来自 Segmentation 模型 (https://github.com/qubvel/segmentation_models.pytorch)。当我运行下面的代码时，我得到

“TracerWarning：将张量转换为 Python 布尔值可能会导致 轨迹不正确。我们无法记录Python的数据流 values，所以这个值在未来会被当作一个常量。 “

这怎么能解决？

input = torch.rand(1,3,512,512)

model = torch.load("saved_model.pth")
torch.onnx.export(model,input,"model.onnx",opset_version=13)

Answer 1

在我的例子中，它发生是因为有一些变量从整数变为张量