小编use*_*935的帖子

如何使用 mmconvert 将 tensorflow 模型(InceptionResnetV2 pb 文件)转换为 pytorch 模型?

Platform : ubuntu 16.04
Python version: 3.5.2
mmdnn version : 0.2.5
Source framework with version : Tensorflow 1.14.0 with GPU
Destination framework with version : Pytorch 1.1.0
Run Code Online (Sandbox Code Playgroud)

预训练模型路径:

初始 resnet v2 模型(tensorflow)(来自https://github.com/tensorflow/models/tree/master/research/slim

运行脚本:

mmconvert -sf tensorflow -iw inception_resnet_v2_jsy.pb --inNodeName input --inputShape 299,299,3 --dstNodeName InceptionResnetV2/Logits/Logits/BiasAdd -df pytorch -om tf_to_pytorch_inception_resnet_v2.pth
Run Code Online (Sandbox Code Playgroud)

你好,

我试图将 tensorflow 模型(inception resnet v2 的 pb 文件)转换为 pytorch 模型以使用 mmdnn。

我使用 pb 文件 (resnet_v1_50, inception_v3) 获得了 2 个模型的成功结果,但是当我尝试转换 inception_resnet_v2 时,出现以下错误。

有没有人有一些想法来解决它们或解释这些问题?

Error logs.


IR network structure is saved …
Run Code Online (Sandbox Code Playgroud)

python tensorflow pytorch

5
推荐指数
1
解决办法
1553
查看次数

标签 统计

python ×1

pytorch ×1

tensorflow ×1