search

DefTruth / lite.ai.toolkit

🛠 A lite C++ toolkit of awesome AI models, support ONNXRuntime, MNN, TNN, NCNN and TensorRT.
https://github.com/DefTruth/lite.ai.toolkit
GNU General Public License v3.0
3.59k stars 684 forks source link

请问onnx模型转TNN模型命令参数是怎样的？ #348

Closed zhongwenkun886 closed 2 years ago

zhongwenkun886 commented 2 years ago

你好，我重新训练人脸检测算法scrfd， 然后转换到onnx模型，然后用以下命令参数转换到TNN模型，我用转换后的模型替换本项目这里的人脸检测scrfd TNN模型（本项目的scrfd TNN模型能正常跑通输出正确结果），但程序输出0个人脸（输入图片有很多人脸）。 然后我又下载了你放到百度网盘上的模型文件 scrfd_500m_bnkps_shape640x640.onnx， 然后用下面命令参数转换到TNN模型，也是输出0个人脸。请问是转换命令参数不对吗？谢谢！

docker run --volume=$(pwd):/workspace -it tnn-convert:latest python3 ./converter.py onnx2tnn \ /workspace/scrfd_500m_bnkps_shape640x640.onnx \ -optimize \ -v v3.0 \ -in "input.1:1,640,640,3"

zhongwenkun886 commented 2 years ago

已解决，应该为 -in "input.1:1,3,640,640"