Closed zhongwenkun886 closed 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"
已解决,应该为 -in "input.1:1,3,640,640"
你好,我重新训练人脸检测算法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"