onnx
文章平均质量分 59
昌山小屋
917
展开
-
【模型转换】onnx删除并新增节点
仍然以【模型转化】修改onnx节点属性这篇文章中的例子为例来学习onnx的基本操作。这次不再修改Squeeze节点的属性(attribute),而是删除原有Squeeze节点,再在原来位置添加一个新的Squeeze节点。1.定位要操作的节点import onnx ...原创 2022-03-28 22:16:03 · 5565 阅读 · 2 评论 -
【模型转化】修改onnx节点属性
在之前的一篇文章【模型转换】onnx转tensorrt报错:Attribute not found: axes中提到squeeze操作在不明确指定axes参数时onnx转tensorrt会报错。解决办法也很简单,因为我是先将pytorch转onnx,再转的tensorrt。pytorch网络结构中,给squeeze操作指定好axes参数再重新生成onnx即可。实际上我们还可以借助onnx的helper功能直接修改onnx节点属性来解决这个问题。 例如,我要将卷积层的输出[1x64x...原创 2022-03-28 15:25:02 · 7402 阅读 · 3 评论