TNN icon indicating copy to clipboard operation
TNN copied to clipboard

convert onnx to tnn failed

Open lxb1226 opened this issue 3 years ago • 1 comments

1. 使用环境(environment)

  • OS: ubuntu
  • OS Version: Ubuntu 18.04.5 LTS

2. Github版本

  • branch:
  • commit(optional):

3. 详细描述bug 情况 (Describe the bug) 我使用pytorch在window上训练了一个dnn模型,然后通过torch.onnx.export函数转换成了onnx模型,在之后转换成tnn模型的时候显示转换失败。从onnx转换到tnn工具是我在wsl2的ubuntu18.04上进行编译的。

  • issue type: 模型转换失败(converter failed)
  • original model: Caffe/ONNX/TensorFlow/TensorFlowLite 原始的pytorch模型: dnn_epoch2.zip 转换后的onnx模型: dnn2.zip

4. 日志(Log) python converter.py onnx2tnn /mnt/e/workspace/GHT/code/new_VAD/dnn2.onnx -in input:1,12 -v v1.0.0

----------  convert model, please wait a moment ----------

Converter ONNX to TNN Model...

Converter ONNX to TNN check_onnx_dim...

Converter ONNX to TNN check_onnx_dim...

Converter ONNX to TNN model failed!

5. 截图(Screenshots) If applicable, add screenshots to help explain your problem. image

lxb1226 avatar Dec 26 '21 07:12 lxb1226

应该是tools\convert2tnn\onnx_converter\onnx2tnn.py 中 这两行去执行cmd时的路径问题 work_dir = "../../onnx2tnn/onnx-converter/" result = cmd.run(command, work_dir=work_dir)

xr-Yang avatar Jul 13 '22 03:07 xr-Yang