转换onnx_PyTorch 到 ONNX 到 Tensorflow 到 TFLite的坑

电玩女神 2023-01-06 01:20 491阅读 0赞

## .view() ##

在 Pytorch 中，经常会对某个具体的 Tensor 做变换的处理。但在 PyTorch 转换到 ONNX 的过程中，例如

fea = fea.view(1, -1, 1, 1)

ONNX 会错误将 view 的纬度扩张变成 Reshape 节点，如下图所示

![db00e9d74ed92af125b89c71cbe1da2a.png][]

但是在 Reshape 节点转换到 Tensorflow 的时候，Reshape 节点就变成了 Where 节点和若干节点的组合以防止出现纬度扩张的时候的 Bug，如下图所示

![a2825dee2f9f4dc528a43734029faa40.png][]

虽然这样子处理貌似稳妥，但是在后续 TF 模型转换到 TFLite 的时候，TFLite 是不支持 Where 节点的。在翻阅 PyTorch Discuss 和 Stackoverflow 的众多讨论后发现大部分人也都是在这里弃坑。

为了避免这个自动转换的问题，这里正确的处理方式为

fea = fea.unsqueeze(-1).unsqueeze(-1)

在修改代码后，我们得到的节点再也不是 Reshape 而是两个 Sequeeze，如下图所示

![4fe0db2c404286f38c4be035c1e108b7.png][]

对应的节点在转换到 Tensorflow的时候，也自动被正确转换成 TFLite 能处理的节点，如下图所示

![c7153aa33adabec6250c837c2838d965.png][]

[db00e9d74ed92af125b89c71cbe1da2a.png]: /images/20221119/b1eac537432145fc9f3858cfbd0f596d.png
[a2825dee2f9f4dc528a43734029faa40.png]: /images/20221119/a7c06bf2229e4a0596477b42b24c72d2.png
[4fe0db2c404286f38c4be035c1e108b7.png]: /images/20221119/5b2f2b54be21436f93e3061df0ddfb8c.png
[c7153aa33adabec6250c837c2838d965.png]: https://img-blog.csdnimg.cn/img_convert/c7153aa33adabec6250c837c2838d965.png

发表评论取消回复

表情：

评论列表（有 0 条评论，491人围观）

还没有评论，来说两句吧...

相关阅读

相关 Tensorflow2.0(Keras)转换TFlite

Tensorflow 2.0(Keras)转换TFlite 目录 Tensorflow 2.0(Keras)转换TFlite 1. TensorFlow Lite 指

古城微笑少年丶/ 2023年02月24日 13:26/ 0 赞/ 1 阅读

相关转换onnx_PyTorch 到 ONNX 到 Tensorflow 到 TFLite的坑

.view() 在 Pytorch 中，经常会对某个具体的 Tensor 做变换的处理。但在 PyTorch 转换到 ONNX 的过程中，例如 fea = fe

电玩女神/ 2023年01月06日 01:20/ 0 赞/ 492 阅读

相关 Python: 从PYTORCH导出模型到ONNX，并使用ONNX运行时运行它

Python: 从PYTORCH导出模型到ONNX，并使用ONNX运行时运行它本教程我们将描述如何将PyTorch中定义的模型转换为ONNX格式，然后使用ONNX运行时

比眉伴天荒/ 2022年11月09日 00:56/ 0 赞/ 766 阅读

相关【目标检测】yolov5模型转换从pytorch到onnx到openvino（部署方式）

1.环境 yolov5环境中有onnx就可以，可以在另外的环境中单独安装openvino。 yolov5的环境 + onnx onnx==1.9.0

待我称王封你为后i/ 2022年10月17日 04:08/ 0 赞/ 819 阅读

相关【C/C++】:string到double和string到int的转换

直接用我下的下面两个函数即可,这是我集成系统的方法，也可以自己使用位数拼接的方法来写这么一个函数，我之前就写过一个，见博客地址： [http://blog.csdn.net

约定不等于承诺〃/ 2022年09月22日 01:29/ 0 赞/ 154 阅读

相关 flask_socketio踩到的坑

需求：读视频里有人像图片的帧，拿到图片帧，传递给其他同事封装好的算法，获取去隐私后坐标信息。实现方法： python服务读取摄像头，拿到摄像头有人像的图片帧，

清疚/ 2022年09月10日 05:16/ 0 赞/ 219 阅读

相关 long 转换到 char*

转摘自：http://hi.baidu.com/lethalboy/blog/item/cac640a0e436baa7caefd025.html itoa ltoa ult

秒速五厘米/ 2022年07月16日 16:57/ 0 赞/ 271 阅读

相关 TensorFlow下载文件到当前目录

简述当使用TensorFlow的时候，默认下载文件，如果是windows会进入到C盘的用户界面内。如果是linux用户，会进入到root目录下。解决办法

朴灿烈づ我的快乐病毒、/ 2022年05月14日 05:16/ 0 赞/ 309 阅读

相关 JSX到JavaScript的转换

可能初次接触React的人会惊讶于为什么可以在JS文件中写HTML的语句，例如`return <div>hello</div>`。其实这种语法就是JSX，它实际上是一种语法

布满荆棘的人生/ 2021年10月30日 02:38/ 0 赞/ 442 阅读

相关使用BigDecimal踩到的坑

今天在敲代码的时候遇到了一个BigDecimal的问题，代码如下： Test testa; Test testb = com.alibaba.fast

ゝ一纸荒年。/ 2021年09月24日 12:22/ 0 赞/ 383 阅读