onnx KeyError: ‘upsample

onnx KeyError: 'upsample_bilinear2d'

解决方法：把上采样的bilinear换成可以支持的。

比如：

import torch.nn.functional as funtion

funtion.interpolate(P3_out, size=P4_td.shape[2:], mode="nearest"))

专门用于2D数据的双线性插值算法：

torch.nn.UpsamplingBilinear2d

CLASS torch.nn.UpsamplingBilinear2d(size=None, scale_factor=None)

size在创建层的时候就指定了，但是预测时，如果动态分辨率，size没法动态改。

但是如果输入是动态分辨率，不知道这里的size应该是多少。
参见：

torch 双线性上采样_jacke121的专栏-CSDN博客_双线性上采样

onnx KeyError: ‘upsample_bilinear2d‘相关推荐

RuntimeError: ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d
这是pth->onnx模型转换时报的错,解决办法是增加opset_version=11: torch.onnx.export(model, (dummy_input, ), 'model.onn ...
ONNX系列四 --- 使用ONNX使TensorFlow模型可移植
目录 TensorFlow简介安装和导入转换器快速浏览模型将TensorFlow模型转换为ONNX 摘要和后续步骤参考文献下载源547.1 KB 系列文章列表如下: ONNX系列一 --- ...
RuntimeError: ONNX export failed: Couldn‘t export operator aten::to
torch.onnx.export(model, input, onnx_path) #add:opset_version=11 修改为: torch.onnx.export(model, input ...
关于在SNPE平台上进行ONNX模型转换DLC模型
Onnx模型转化DLC模型简介在snpe平台上,将onnx模型转换为dlc模型目录 snpe平台介绍 snpe平台与onnx配置 onnx模型转换dlc 模型量化关于1.38版本SNPE部署时 ...
ONNX MLIR应用示例（含源码链接）
ONNX MLIR应用示例(含源码链接) 开放式神经网络交换在MLIR中的实现 (http://onnx.ai/onnx-mlir/). Prebuilt Containers 开始使用ONNX-ML ...
ONNX MLIR方法
ONNX MLIR方法 MLIR中的开放式神经网络交换实现. Prerequisites gcc >= 6.4 libprotoc >= 3.11.0 cmake >= 3.15.4 ...
使用Relay部署编译ONNX模型
使用Relay部署编译ONNX模型本文介绍如何使用Relay部署ONNX模型的入门. 首先,必须安装ONNX软件包. 一个快速的解决方案是安装protobuf编译器,然后 pip install o ...
使用ONNX将模型转移至Caffe2和移动端
使用ONNX将模型转移至Caffe2和移动端本文介绍如何使用 ONNX 将 PyTorch 中定义的模型转换为 ONNX 格式,然后将其加载到 Caffe2 中.一旦进入 Caffe2, 就可以运行 ...
ONNX 实时graph优化方法
ONNX 实时graph优化方法 ONNX实时提供了各种图形优化来提高模型性能.图优化本质上是图级别的转换,从小型图简化和节点消除,到更复杂的节点融合和布局优化. 图形优化根据其复杂性和功能分为几个类 ...

onnx KeyError: ‘upsample_bilinear2d‘

onnx KeyError: ‘upsample_bilinear2d‘相关推荐

最新文章

热门文章