Unsupported ONNX opset version: 11

最新推荐文章于 2024-07-25 12:46:18 发布

AI算法网奇

最新推荐文章于 2024-07-25 12:46:18 发布

阅读量1.4w

点赞数 1

分类专栏： onnx

本文链接：https://blog.csdn.net/jacke121/article/details/104503121

版权

onnx 专栏收录该内容

96 篇文章 25 订阅

订阅专栏

Unsupported ONNX opset version: 11

估计是pytorch不支持，重新安装pytorch2.0可能能解决。

比如：

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

torch.onnx.export(net, input, onnx_name, verbose=False,opset_version=11)

opset_version=11 参数设置成功后，onnx能生成，但是转mnn不支持。

原因：一遍都是上采样引起的：

  up3 = F.interpolate(output3_, size=[output2_.size(2), output2_.size(3)], mode="nearest")
        output2 = output2_ + up3
        output2 = self.merge2(output2)

        up2 = F.interpolate(output2, size=[output1_.size(2), output1_.size(3)], mode="nearest")
        output1 = output1_ + up2
        output1 = self.merge1(output1)