TNN
TNN copied to clipboard
convert onnx to tnn failed
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.
应该是tools\convert2tnn\onnx_converter\onnx2tnn.py 中 这两行去执行cmd时的路径问题 work_dir = "../../onnx2tnn/onnx-converter/" result = cmd.run(command, work_dir=work_dir)